From 3d2c534933b356dc313a620639a713cb940dc756 Mon Sep 17 00:00:00 2001
From: Taesung Park <taesung_park@berkeley.edu>
Date: Sun, 10 Dec 2017 23:00:17 -0800
Subject: added script for downloading pretrained CycleGAN models

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

(limited to 'pretrained_models')

diff --git a/pretrained_models/download_cyclegan_model.sh b/pretrained_models/download_cyclegan_model.sh
new file mode 100644
index 0000000..91f0021
--- /dev/null
+++ b/pretrained_models/download_cyclegan_model.sh
@@ -0,0 +1,13 @@
+FILE=$1
+
+echo "Note: available models are horse2zebra, zebra2horse"
+
+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/models/$FILE.pth
+
+wget -N $URL -O $MODEL_FILE
+
+
-- 
cgit v1.2.3-70-g09d2