diff options
| -rw-r--r-- | pretrained_models/download_pix2pix_model.sh | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/pretrained_models/download_pix2pix_model.sh b/pretrained_models/download_pix2pix_model.sh new file mode 100644 index 0000000..ee51a43 --- /dev/null +++ b/pretrained_models/download_pix2pix_model.sh @@ -0,0 +1,13 @@ +FILE=$1 + +echo "Note: available models are edges2shoes, sat2map, and facades_label2photo" + +echo "Specified [$FILE]" + +mkdir -p ./checkpoints/${FILE}_pretrained +MODEL_FILE=./checkpoints/${FILE}_pretrained/latest_net_G.pth +URL=https://people.eecs.berkeley.edu/~taesung_park/pytorch-CycleGAN-and-pix2pix/pix2pix_models/$FILE.pth + +wget -N $URL -O $MODEL_FILE + + |