diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-12-10 23:00:17 -0800 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-12-10 23:00:17 -0800 |
| commit | 3d2c534933b356dc313a620639a713cb940dc756 (patch) | |
| tree | a70550cbd4ac050073df615635d65bd3ceda3fda | |
| parent | b6f5966eb8224dfc7be68b1b67a87f006e42730d (diff) | |
added script for downloading pretrained CycleGAN models
| -rw-r--r-- | pretrained_models/download_cyclegan_model.sh | 13 |
1 files changed, 13 insertions, 0 deletions
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 + + |