summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTaesung Park <taesung_park@berkeley.edu>2017-10-10 01:44:01 -0700
committerTaesung Park <taesung_park@berkeley.edu>2017-10-10 01:44:01 -0700
commit5e0f7d6980ed1a1aaac8593351028d320e5f0a94 (patch)
tree5a9e67896a74318b8da1027e9617c8c51fd2c3a5
parentae9042ee8d913f15395117afb8eea9bdc6f72499 (diff)
added script to download pretrained pix2pix models
-rw-r--r--pretrained_models/download_pix2pix_model.sh13
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
+
+