diff --git a/scripts/download_model.sh b/scripts/download_model.sh new file mode 100644 index 0000000..4d2dccc --- /dev/null +++ b/scripts/download_model.sh @@ -0,0 +1,19 @@ +mkdir checkpoints +cd checkpoints + +wget https://storage.googleapis.com/self-distilled-stylegan/lions_512_pytorch.pkl +mv lions_512_pytorch.pkl stylegan2_lions_512_pytorch.pkl + +wget https://storage.googleapis.com/self-distilled-stylegan/dogs_1024_pytorch.pkl +mv dogs_1024_pytorch.pkl stylegan2_dogs_1024_pytorch.pkl + +wget https://storage.googleapis.com/self-distilled-stylegan/horses_256_pytorch.pkl +mv horses_256_pytorch.pkl stylegan2_horses_256_pytorch.pkl + +wget https://storage.googleapis.com/self-distilled-stylegan/elephants_512_pytorch.pkl +mv elephants_512_pytorch.pkl stylegan2_elephants_512_pytorch.pkl + +wget https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan2/versions/1/files/stylegan2-ffhq-512x512.pkl +wget https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan2/versions/1/files/stylegan2-afhqcat-512x512.pkl +wget http://d36zk2xti64re0.cloudfront.net/stylegan2/networks/stylegan2-car-config-f.pkl +wget http://d36zk2xti64re0.cloudfront.net/stylegan2/networks/stylegan2-cat-config-f.pkl diff --git a/scripts/gui.sh b/scripts/gui.sh new file mode 100644 index 0000000..5eb68e3 --- /dev/null +++ b/scripts/gui.sh @@ -0,0 +1,11 @@ +python visualizer_drag.py \ + checkpoints/stylegan2_lions_512_pytorch.pkl \ + checkpoints/stylegan2-ffhq-512x512.pkl \ + checkpoints/stylegan2-afhqcat-512x512.pkl \ + checkpoints/stylegan2-car-config-f.pkl \ + checkpoints/stylegan2_dogs_1024_pytorch.pkl \ + checkpoints/stylegan2_horses_256_pytorch.pkl \ + checkpoints/stylegan2-cat-config-f.pkl \ + checkpoints/stylegan2_elephants_512_pytorch.pkl \ + checkpoints/stylegan_human_v2_512.pkl \ + checkpoints/stylegan2-lhq-256x256.pkl