From 55b2cd3b31f3e0dd3e4c4cfcc5ae90a9faefcef6 Mon Sep 17 00:00:00 2001 From: HieuPhan33 <hieup007@gmail.com> Date: Thu, 14 Dec 2023 18:43:10 +1100 Subject: [PATCH] train.sh --- .gitignore | 6 ++---- test.sh | 7 +++++++ train.sh | 13 +++++++++++++ 3 files changed, 22 insertions(+), 4 deletions(-) create mode 100755 test.sh create mode 100755 train.sh diff --git a/.gitignore b/.gitignore index 42ea1ed..3ad90e7 100644 --- a/.gitignore +++ b/.gitignore @@ -1,6 +1,4 @@ checkpoints* -AttentionGAN-v1* -AttentionGAN-geo *pycache* sam wandb @@ -10,5 +8,5 @@ scripts/GAN_Metrics-Tensorflow/*gz results .vscode wandb -train.sh -test.sh \ No newline at end of file +# train.sh +# test.sh \ No newline at end of file diff --git a/test.sh b/test.sh new file mode 100755 index 0000000..ce69fbe --- /dev/null +++ b/test.sh @@ -0,0 +1,7 @@ +# python test.py --gpu_ids 0 --dataroot /data/data/paediatric-head/processed_img_open \ +# mask_gan --netG att \ +# --checkpoints_dir checkpoints --load_size 150 --pad_size 225 --crop_size 224 --preprocess resize_pad_crop \ + +python test.py --dataroot /data/data/processed-pelvis-mr-ct-mask --gpu_ids 0 --model mask_gan --name hieu_mr_ct_pelvis_attgan --dataset_mode unaligned --no_dropout \ +--norm instance --preprocess none --Aclass A --Bclass B --netG att + diff --git a/train.sh b/train.sh new file mode 100755 index 0000000..3e92531 --- /dev/null +++ b/train.sh @@ -0,0 +1,13 @@ +#CUDA_VISIBLE_DEVICES=0 python train.py --dataroot /data/paediatric-head/processed_img_v2 --gpu_ids 0 --display_id 0 --name cycle-corr --batchSize 64 --fineSize 160 --loadSize 198 +#python train.py --dataroot /data/paediatric-head/processed_img_crop --gpu_ids 1 --display_id 0 --name cycle-structure-amp-crop --batchSize 42 --lambda_co_A 0 --lambda_co_B 0 --lambda_s_A 0.5 --lambda_s_B 0.5 --fineSize 200 --loadSize 224 --niter 50 --niter_decay 50 --serial_batches +#python train.py --dataroot /data/paediatric-head/processed_img_crop --gpu_ids 1 --display_id 0 --name unet-amp-crop --batchSize 128 --lambda_co_A 0 --lambda_co_B 0 --lambda_s_A 0.0 --lambda_s_B 0.0 --fineSize 256 --loadSize 280 --niter 50 --niter_decay 50 --serial_batches --which_model_netG unet_256 + +# hieu-torch attention-mask-shape +# python train.py --dataroot /data/data/paediatric-head/processed_img_open --gpu_ids 1 --display_id 0 --model mask_gan --name mask_gan --netG att \ +# --dataset_mode unaligned --pool_size 50 --no_dropout --amp_level O0 \ +# --norm instance --lambda_A 10 --lambda_B 10 --lambda_identity 0.5 --lambda_mask 1.0 --lambda_shape 0.5 --load_size 150 --pad_size 225 --crop_size 224 --preprocess resize_pad_crop --no_flip \ +# --batch_size 4 --niter 40 --niter_decay 40 --display_freq 1000 --print_freq 1000 --n_attentions 5 + +python train.py --dataroot /data/data/processed-pelvis-mr-ct-mask --gpu_ids 0 --display_id 0 --model mask_gan --name hieu_mr_ct_pelvis_attgan --dataset_mode unaligned --pool_size 50 --no_dropout \ +--norm instance --lambda_mask 0.0 --lambda_co_B 0.0 --preprocess none --opt_level O1 --lambda_A 10 --lambda_B 10 --lambda_identity 0.5 --lambda_shape 0.0 \ +--batch_size 12 --niter 50 --niter_decay 50 --display_freq 1000 --print_freq 1000 --Aclass A --Bclass B --half