diff options
| -rwxr-xr-x | run-gan.sh | 37 | ||||
| -rwxr-xr-x | run-messi.sh | 38 | ||||
| -rwxr-xr-x | run-wood.sh | 37 | ||||
| -rwxr-xr-x | run100k.sh | 32 | ||||
| -rwxr-xr-x | test-all.sh | 24 | ||||
| -rw-r--r-- | test-datasets.sh | 26 |
6 files changed, 194 insertions, 0 deletions
diff --git a/run-gan.sh b/run-gan.sh new file mode 100755 index 0000000..310f7b5 --- /dev/null +++ b/run-gan.sh @@ -0,0 +1,37 @@ +dataset="woodscaled_unaligned" + +#/home/lens/Desktop/dataset-random.pl +# +#python datasets/combine_A_and_B.py \ +# --fold_A "/home/lens/Desktop/thumbs/$dataset/A" \ +# --fold_B "/home/lens/Desktop/thumbs/$dataset/B" \ +# --fold_AB "./datasets/$dataset/" + +python train.py \ + --dataroot "./datasets/$dataset" \ + --name "$dataset" \ + --model cycle_gan \ + --loadSize 256 \ + --fineSize 256 \ + --which_model_netG unet_256 \ + --which_direction AtoB \ + --lambda_B 100 \ + --dataset_mode aligned \ + --no_lsgan --norm batch --pool_size 0 \ + --continue_train + +python test.py \ + --dataroot "/home/lens/Desktop/thumbs/$dataset/A/train/" \ + --name "$dataset" \ + --start_img "/home/lens/Desktop/thumbs/$dataset/A/train/frame_1008.png" \ + --how_many 1000 \ + --model test \ + --aspect_ratio 1.777777 \ + --which_model_netG unet_256 \ + --which_direction AtoB \ + --dataset_mode recursive \ + --loadSize 256 \ + --fineSize 256 \ + --norm batch + + diff --git a/run-messi.sh b/run-messi.sh new file mode 100755 index 0000000..63d9a9a --- /dev/null +++ b/run-messi.sh @@ -0,0 +1,38 @@ +dataset="messi" + +#perl /home/lens/Desktop/pair-dataset.pl +#sleep 1 + +#python datasets/combine_A_and_B.py \ +# --fold_A "/home/lens/Desktop/thumbs/$dataset/A" \ +# --fold_B "/home/lens/Desktop/thumbs/$dataset/B" \ +# --fold_AB "./datasets/$dataset/" +#sleep 1 + +#python train.py \ +# --dataroot "./datasets/$dataset" \ +# --name "$dataset" \ +# --model pix2pix \ +# --loadSize 256 \ +# --fineSize 256 \ +# --which_model_netG unet_256 \ +# --which_direction AtoB \ +# --lambda_B 100 \ +# --dataset_mode aligned \ +# --no_lsgan --norm batch --pool_size 0 + +python test.py \ + --dataroot "/home/lens/Desktop/thumbs/$dataset/A/train/" \ + --name "$dataset" \ + --start_img "/home/lens/Desktop/thumbs/$dataset/A/train/frame_0000.png" \ + --how_many 500 \ + --model test \ + --aspect_ratio 1.777777 \ + --which_model_netG unet_256 \ + --which_direction AtoB \ + --dataset_mode recursive \ + --loadSize 286 \ + --fineSize 256 \ + --norm batch + #--center_crop \ + diff --git a/run-wood.sh b/run-wood.sh new file mode 100755 index 0000000..d053777 --- /dev/null +++ b/run-wood.sh @@ -0,0 +1,37 @@ +dataset="randomcrops5k" + +#/home/lens/Desktop/dataset-random.pl +# +#python datasets/combine_A_and_B.py \ +# --fold_A "/home/lens/Desktop/thumbs/$dataset/A" \ +# --fold_B "/home/lens/Desktop/thumbs/$dataset/B" \ +# --fold_AB "./datasets/$dataset/" +# python train.py \ +# --dataroot "./datasets/$dataset" \ +# --name "$dataset" \ +# --model pix2pix \ +# --loadSize 256 \ +# --fineSize 256 \ +# --which_model_netG unet_256 \ +# --which_direction AtoB \ +# --lambda_B 100 \ +# --dataset_mode aligned \ +# --no_lsgan --norm batch --pool_size 0 \ +# --continue_train + +python test.py \ + --dataroot "/home/lens/Desktop/thumbs/$dataset/A/train/" \ + --name "$dataset" \ + --start_img "/home/lens/Desktop/thumbs/$dataset/A/train/frame_0000.png" \ + --how_many 500 \ + --model test \ + --aspect_ratio 1.777777 \ + --which_model_netG unet_256 \ + --which_direction AtoB \ + --dataset_mode recursive \ + --loadSize 264 \ + --center_crop \ + --fineSize 256 \ + --norm batch + + diff --git a/run100k.sh b/run100k.sh new file mode 100755 index 0000000..1ad9e77 --- /dev/null +++ b/run100k.sh @@ -0,0 +1,32 @@ +dataset="randomcrops100k" + +/home/lens/Desktop/dataset-random.pl + +python datasets/combine_A_and_B.py \ + --fold_A "/home/lens/Desktop/thumbs/$dataset/A" \ + --fold_B "/home/lens/Desktop/thumbs/$dataset/B" \ + --fold_AB "./datasets/$dataset/" + +python train.py \ + --dataroot "./datasets/$dataset" \ + --name "$dataset" \ + --model pix2pix \ + --which_model_netG unet_256 \ + --which_direction AtoB \ + --lambda_B 100 \ + --dataset_mode aligned \ + --no_lsgan --norm batch --pool_size 0 + +python test.py \ + --dataroot "/home/lens/Desktop/thumbs/$dataset/A/train/" \ + --name "$dataset" \ + --start_img "/home/lens/Desktop/thumbs/$dataset/A/train/frame_0000.png" \ + --how_many 2000 \ + --model test \ + --aspect_ratio 1.777777 \ + --which_model_netG unet_256 \ + --which_direction AtoB \ + --dataset_mode recursive \ + --norm batch + + diff --git a/test-all.sh b/test-all.sh new file mode 100755 index 0000000..675966e --- /dev/null +++ b/test-all.sh @@ -0,0 +1,24 @@ +declare -a datasets=("/home/lens/Desktop/skull/") +declare -a checkpoints=("messi" "randomcrops100k" "randomcrops_30000_4" "randomcrops5k" "woodscaled" "woodscaled_4") + +for ds in "${datasets[@]}" +do + +bs=$(basename $ds) + +for i in "${checkpoints[@]}" +do + echo $ds + echo $i + + python test.py \ + --dataroot $ds \ + --name $i --mov "${bs}_warp_${i}" \ + --model test --dataset_mode single --experiment woodwarp \ + --which_model_netG unet_256 --which_direction AtoB \ + --loadSize 256 --fineSize 256 --norm batch --how_many 10000 + +done + +done + diff --git a/test-datasets.sh b/test-datasets.sh new file mode 100644 index 0000000..3629170 --- /dev/null +++ b/test-datasets.sh @@ -0,0 +1,26 @@ +declare -a datasets=("blended_wood_canny_leaves" "blended_wood_canny_typical" "blended_wood_canny_more_edges_poorly_trained") +declare -a checkpoints=("messi" "randomcrops100k" "randomcrops_30000_4" "randomcrops5k" "woodscaled" "woodscaled_4") + +basedir="/home/lens/code/torch-warp/" + +for ds in "${datasets[@]}" +do + +bs=$(basename $ds) + +for i in "${checkpoints[@]}" +do + echo $ds + echo $i + + python test.py \ + --dataroot "${basedir}${ds}/256/"\ + --name $i --mov "${bs}_warp_${i}" \ + --model test --dataset_mode single --experiment woodwarp \ + --which_model_netG unet_256 --which_direction AtoB \ + --loadSize 256 --fineSize 256 --norm batch --how_many 10000 + +done + +done + |