Merge pull request #72 from bulutthecat/main

Added windows support for GUI
2 years ago · d1ed11c952
parent ccd84ffab9 9059a77f8e
commit d1ed11c952
2 changed files with 16 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -3,4 +3,7 @@ torchvision
 Ninja
 gradio
 huggingface_hub
-hf_transfer
+hf_transfer
+pyopengl
+imgui
+glfw
--- a/scripts/gui.bat
+++ b/scripts/gui.bat
@ -0,0 +1,12 @@
+@echo off
+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