diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-10-10 01:44:01 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-10-10 01:44:01 -0700 |
| commit | 5e0f7d6980ed1a1aaac8593351028d320e5f0a94 (patch) | |
| tree | 5a9e67896a74318b8da1027e9617c8c51fd2c3a5 | |
| parent | ae9042ee8d913f15395117afb8eea9bdc6f72499 (diff) | |
added script to download pretrained pix2pix models
| -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 + + |
