From 5e0f7d6980ed1a1aaac8593351028d320e5f0a94 Mon Sep 17 00:00:00 2001
From: Taesung Park <taesung_park@berkeley.edu>
Date: Tue, 10 Oct 2017 01:44:01 -0700
Subject: added script to download pretrained pix2pix models

---
 pretrained_models/download_pix2pix_model.sh | 13 +++++++++++++
 1 file changed, 13 insertions(+)
 create mode 100644 pretrained_models/download_pix2pix_model.sh

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
+
+
-- 
cgit v1.2.3-70-g09d2