add precomputed directions

README.md

@@ -1,13 +0,0 @@
----
-title: StyleMC Demo
-emoji: 🏢
-colorFrom: gray
-colorTo: pink
-sdk: gradio
-sdk_version: 3.14.0
-app_file: app.py
-pinned: false
-license: apache-2.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import legacy
 import dnnlib
@@ -14,6 +15,7 @@ landmarks_path = "./pretrained/shape_predictor_68_face_landmarks.dat"
 e4e_embedder = psp_wrapper.psp_encoder(psp_encoder_path, landmarks_path)
 G_ffhq_path = "./pretrained/ffhq.pkl"
 G_metfaces_path = "./pretrained/metfaces.pkl"
+direction_folder = "./assets/directions/"
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -34,6 +36,14 @@ FOOTER = 'This space is built by <a href = "https://github.com/catlab-team">Catl
 direction_map = {}
 direction_list = []
 
+directions = [f for f in os.listdir(direction_folder) if f.endswith(".npz")]
+for d in directions:
+    with np.load(direction_folder + d) as data:
+        dir_name = f.split(".npz")[0]
+        direction_list.append(data["s"])
+        direction_map[dir_name] = {"direction": data["s"], "stylegan_type": "FFHQ"} 
+
+
 def add_direction(prompt, stylegan_type, id_loss_w):
     new_dir_name = prompt+" "+stylegan_type+" w_id_loss"+str(id_loss_w)
     if (prompt != None) and (new_dir_name not in direction_list):
@@ -41,7 +51,7 @@ def add_direction(prompt, stylegan_type, id_loss_w):
         direction = find_direction.find_direction(G_dict[stylegan_type], prompt)
         print(f"new direction calculated with {stylegan_type} and id loss weight = {id_loss_w}")
         direction_list.append(new_dir_name)
-        direction_map[new_dir_name]={"direction":direction, "stylegan_type":stylegan_type}
+        direction_map[new_dir_name] = {"direction":direction, "stylegan_type":stylegan_type}
 
     return gr.Radio.update(choices=direction_list, value=None, visible=True)
 

assets/directions/photo_of_a_face_with_beard.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e1b9a508339550ca0d479ac09db76d600db173e8fec6a40b7b82c240b8889c6
+size 53504

assets/directions/photo_of_a_face_with_blonde_hair.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cbf362827734953aa1b67bee79928b53d98342a439165c73926c65189be3389
+size 53504

assets/directions/photo_of_a_face_with_curly_hair.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:827407ced6a64889d368f0acabb94d24ee88be3efe77848e77b771db7e65e6ac
+size 53504

assets/directions/photo_of_a_face_with_long_hair.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:737ce411fadf60d40f5384007fbd556335f3800b8a7b006e459adbfbb90f462a
+size 53504

assets/directions/photo_of_a_frowning_face.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23d7248ac893c249eb44d11bbe79f91ad469049a20ade43cfba3e55cdad02283
+size 53504

assets/directions/photo_of_a_happy_face.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b3990c63d44e52cbe8d73f40f07a0b1dbb9d309cbac032427d316698d4fa94e
+size 53504

assets/directions/photo_of_a_relieved_face.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0bcde1467755880f0df47839031f10c82eb93b0ecf03e8e217360c396001cad
+size 53504

assets/directions/photo_of_a_tanned_skin_face.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56613c2f6a3844681370e1939f20dea62b339a78c4a48380f27244e196e23dd1
+size 53504

assets/directions/photo_of_an_excited_face.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c4d7b528ac1e3c9391a826470b61dfa8e0db8a44adcd9b52a8ab027c37c4259d
+size 53504