forked from NVlabs/stylegan2-ada-pytorch
-
Notifications
You must be signed in to change notification settings - Fork 3
/
run_experiments.sh
executable file
·83 lines (67 loc) · 2.59 KB
/
run_experiments.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
echo "Running Blending Experiments"
CUDA_VISIBLE_DEVICES=0
# CAT_CHKPT_PATH="training_runs/00004-afhqcat256-mirror-auto2-resumeffhq256/network-snapshot-000560.pkl"
# DOG_CHKPT_PATH="training_runs/00007-afhqdog256-mirror-auto1-resumeffhq256/network-snapshot-000640.pkl"
# WILD_CHKPT_PATH="training_runs/00009-afhqwild256-mirror-auto1-resumeffhq256/network-snapshot-000600.pkl"
# CARTOON_CHKPT_PATH="training_runs/00008-cartoon256-mirror-auto1-resumeffhq256/network-snapshot-000640.pkl"
# PRETRAINED_FFHQ="https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada-pytorch/pretrained/transfer-learning-source-nets/ffhq-res256-mirror-paper256-noaug.pkl"
PRETRAINED_FFHQ="checkpoints/ffhq_chkpt_256.pkl"
CAT_CHKPT_PATH="checkpoints/cat_chkpt_256.pkl"
DOG_CHKPT_PATH="checkpoints/dog_chkpt_256.pkl"
WILD_CHKPT_PATH="checkpoints/wild_chkpt_256.pkl"
CARTOON_CHKPT_PATH="checkpoints/cartoon_chkpt_256.pkl"
echo "Faces -> Cats"
python stylegan_blending.py \
--network1 $PRETRAINED_FFHQ \
--network2 $CAT_CHKPT_PATH \
--outdir experiments_faces_to_cats --dim 256
echo "Cats -> Faces"
python stylegan_blending.py \
--network2 $PRETRAINED_FFHQ \
--network1 $CAT_CHKPT_PATH \
--outdir experiments_cats_to_faces --dim 256
echo "Faces -> Dogs"
python stylegan_blending.py \
--network1 $PRETRAINED_FFHQ \
--network2 $DOG_CHKPT_PATH \
--outdir experiments_faces_to_dogs --dim 256
echo "Dogs -> Faces"
python stylegan_blending.py \
--network2 $PRETRAINED_FFHQ \
--network1 $DOG_CHKPT_PATH \
--outdir experiments_dogs_to_faces --dim 256
echo "Faces -> Wildlife"
python stylegan_blending.py \
--network1 $PRETRAINED_FFHQ \
--network2 $WILD_CHKPT_PATH \
--outdir experiments_faces_to_wild --dim 256
echo "Wildlife -> Faces"
python stylegan_blending.py \
--network2 $PRETRAINED_FFHQ \
--network1 $WILD_CHKPT_PATH \
--outdir experiments_wild_to_faces --dim 256
echo "Faces -> CartoonFace"
python stylegan_blending.py \
--network1 $PRETRAINED_FFHQ \
--network2 $CARTOON_CHKPT_PATH \
--outdir experiments_faces_to_cartoon --dim 256
echo "CartoonFace -> Faces"
python stylegan_blending.py \
--network2 $PRETRAINED_FFHQ \
--network1 $CARTOON_CHKPT_PATH \
--outdir experiments_cartoon_to_faces --dim 256
echo "Dogs -> Cats"
python stylegan_blending.py \
--network1 $DOG_CHKPT_PATH \
--network2 $CAT_CHKPT_PATH \
--outdir experiments_dogs_to_cats --dim 256
echo "Wildlife -> Cats"
python stylegan_blending.py \
--network1 $WILD_CHKPT_PATH \
--network2 $CAT_CHKPT_PATH \
--outdir experiments_wild_to_cats --dim 256
echo "Cartoons -> Cats"
python stylegan_blending.py \
--network1 $CARTOON_CHKPT_PATH \
--network2 $CAT_CHKPT_PATH \
--outdir experiments_cartoon_to_cats --dim 256