Diggz10 commited on
Commit
4be5660
·
verified ·
1 Parent(s): 03251a4

Upload 35 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,19 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ stylegan2directions/age.jpg filter=lfs diff=lfs merge=lfs -text
37
+ stylegan2directions/eye_distance.jpg filter=lfs diff=lfs merge=lfs -text
38
+ stylegan2directions/eye_eyebrow_distance.jpg filter=lfs diff=lfs merge=lfs -text
39
+ stylegan2directions/eye_ratio.jpg filter=lfs diff=lfs merge=lfs -text
40
+ stylegan2directions/eyes_open.jpg filter=lfs diff=lfs merge=lfs -text
41
+ stylegan2directions/gender.jpg filter=lfs diff=lfs merge=lfs -text
42
+ stylegan2directions/lip_ratio.jpg filter=lfs diff=lfs merge=lfs -text
43
+ stylegan2directions/mouth_open.jpg filter=lfs diff=lfs merge=lfs -text
44
+ stylegan2directions/mouth_ratio.jpg filter=lfs diff=lfs merge=lfs -text
45
+ stylegan2directions/nose_mouth_distance.jpg filter=lfs diff=lfs merge=lfs -text
46
+ stylegan2directions/nose_ratio.jpg filter=lfs diff=lfs merge=lfs -text
47
+ stylegan2directions/nose_tip.jpg filter=lfs diff=lfs merge=lfs -text
48
+ stylegan2directions/pitch.jpg filter=lfs diff=lfs merge=lfs -text
49
+ stylegan2directions/roll.jpg filter=lfs diff=lfs merge=lfs -text
50
+ stylegan2directions/smile.jpg filter=lfs diff=lfs merge=lfs -text
51
+ stylegan2directions/yaw.jpg filter=lfs diff=lfs merge=lfs -text
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import numpy as np
3
+ import torch
4
+ import dnnlib
5
+ import legacy
6
+
7
+ # Load pretrained StyleGAN2 model
8
+ with open("stylegan2-ffhq-config-f.pkl", "rb") as f:
9
+ G = legacy.load_network_pkl(f)["G_ema"].to(torch.device("cpu"))
10
+
11
+ # Load gender direction vector
12
+ gender_direction = np.load("stylegan2directions/gender.npy")
13
+
14
+ def edit_gender(seed, strength):
15
+ rnd = np.random.RandomState(seed)
16
+ z = torch.from_numpy(rnd.randn(1, G.z_dim)).to(torch.float32)
17
+ w = G.mapping(z, None, truncation_psi=0.5)
18
+ w += torch.from_numpy(gender_direction).to(torch.float32) * strength
19
+ img = G.synthesis(w, noise_mode="const")
20
+ img = (img.clamp(-1, 1) + 1) * 127.5
21
+ img = img.permute(0, 2, 3, 1)[0].cpu().numpy().astype(np.uint8)
22
+ return img
23
+
24
+ gr.Interface(
25
+ fn=edit_gender,
26
+ inputs=[
27
+ gr.Slider(0, 10000, step=1, label="Random Seed"),
28
+ gr.Slider(-5, 5, step=0.1, label="Gender Strength (← Feminine | Masculine →)")
29
+ ],
30
+ outputs=gr.Image(label="Generated Face"),
31
+ title="Gender Editing with StyleGAN2",
32
+ description="Move the slider to masculinize or feminize the face."
33
+ ).launch()
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ gradio
2
+ torch
3
+ numpy
4
+ dnnlib
stylegan2-ffhq-config-f.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adf127ea7bb8a7788c8bdeda3c9937f7310b669b09ecf799ca53a631ff46948d
3
+ size 381673535
stylegan2directions/age.jpg ADDED

Git LFS Details

  • SHA256: 9f363b2a64f9dbe63d5308af59d47e9d8e702eef335023f7abb7f4e5c7b6a1d4
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/age.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72095858d9f441494d46742639da43b17c282bbd48add6f5eca143073d725986
3
+ size 73856
stylegan2directions/eye_distance.jpg ADDED

Git LFS Details

  • SHA256: c1ab016e9d8840a806e41fde1eee1eac1d966a29a3b1ff751647b829f88e9ab8
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/eye_distance.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c679f7405f8277ef791aa22fd8cb7c728bd0ac210fcd9664cf57c3138c9c4ef6
3
+ size 73856
stylegan2directions/eye_eyebrow_distance.jpg ADDED

Git LFS Details

  • SHA256: 64199f5e1c4a8825e0be1ec767d63e7c77fffea9236b7a224d409fcf76e69b89
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
stylegan2directions/eye_eyebrow_distance.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a13002c0ad517fdcffe1688e3a3b44a1b8fab0a71769e609b6de43a20550edf0
3
+ size 73856
stylegan2directions/eye_ratio.jpg ADDED

Git LFS Details

  • SHA256: 9bd75f7e26b8152ec197edd02e33141945a687f33e5a4e8dcbe0fdfca0c35e41
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/eye_ratio.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77a7aa387c857eea20380a530626e48ba63de1ad33621c37319020a823be7446
3
+ size 73856
stylegan2directions/eyes_open.jpg ADDED

Git LFS Details

  • SHA256: a656cdd8d508ee06d03e2cd17857b26737039d012af5cc14c261654ad7d31758
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/eyes_open.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6115c94f42449aaa1f7064171f32bdc6d6c92aba4db7905fdbbd31a1483fde90
3
+ size 73856
stylegan2directions/gender.jpg ADDED

Git LFS Details

  • SHA256: 44b5d3d4272ddea6d4c680e2ce9fbe830f18b2c7a707cee1794b9299c08d6c3a
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
stylegan2directions/gender.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1ac40508d9f5b7df499378f60766d93fe4e5f5461b63fff8ed14233e9b15a9e
3
+ size 73856
stylegan2directions/lip_ratio.jpg ADDED

Git LFS Details

  • SHA256: 2e7f7fdb6aadeca43b01622c5f6024f173fc494baa130e1e212ae505d04ffb4f
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/lip_ratio.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3ee87e7a175666437d722dbb191cd1b73fd5ce02955f058cbe280878bd529b0
3
+ size 73856
stylegan2directions/mouth_open.jpg ADDED

Git LFS Details

  • SHA256: 02bdd89954287d4c6e441b76b8a90f83c01fde756b4ff1fb9cc5cdcc27e173f7
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/mouth_open.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4852b360bbc70380f2151e5d7204a895ccf8d22d6a951dbb6f65264774dab39
3
+ size 73856
stylegan2directions/mouth_ratio.jpg ADDED

Git LFS Details

  • SHA256: 39eba119a1180230ac588f88bac04b960aea208db55eeb2c6ff1f52be20edfb6
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/mouth_ratio.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a897cc731a213a3ec62869181a30d31d72a2761340c58d0f24e073a5a67be28
3
+ size 73856
stylegan2directions/nose_mouth_distance.jpg ADDED

Git LFS Details

  • SHA256: 02aed81afb979039e95d59999e0839d2f16521e3fbcbfe278e2fbe4d57ba54a7
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/nose_mouth_distance.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a466d61d2123d64134616656d36ae5f50fa9bf345373f2e51d37fc41d1af30ca
3
+ size 73856
stylegan2directions/nose_ratio.jpg ADDED

Git LFS Details

  • SHA256: c815021e49b9b7ee67cd5a114f80b055b013c83e75aa4293073e74886c962cfd
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
stylegan2directions/nose_ratio.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8866c8cf02f7aad7a13c71c3f6b64f9414a58f43a2d4b8f52982c86b4c1dc2d3
3
+ size 73856
stylegan2directions/nose_tip.jpg ADDED

Git LFS Details

  • SHA256: 1ce946c1e017ae2551255af72e561751f43f3706428e38f2bd9cf2e5fe83c62a
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
stylegan2directions/nose_tip.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aba1e4487b43e8a1389b3c645b83fe032c439f32397c04e24d3dc9c0e63857b0
3
+ size 73856
stylegan2directions/pitch.jpg ADDED

Git LFS Details

  • SHA256: a7e0ae4db0c96ff2e1140d2f229ca542b7323a52cfebe0dc5b09426042b0b1c3
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
stylegan2directions/pitch.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28fd460a828f56f83c6c22ce4c5251664dab5d7d5da3ad840517c6df9d07c385
3
+ size 73856
stylegan2directions/roll.jpg ADDED

Git LFS Details

  • SHA256: 88a8b50339025ef8259ceb867549c40a9aa44be2d3a4c138cb14ee4e1c1e2769
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
stylegan2directions/roll.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4e0cc381482a79cd8901e0997cb1d4ffdf6c8e597306da32a040f290e0fed88
3
+ size 73856
stylegan2directions/smile.jpg ADDED

Git LFS Details

  • SHA256: 3919f1bbbb9d1a2102a03d2198c156f75c716d56e0bd47219b51a0153b398281
  • Pointer size: 131 Bytes
  • Size of remote file: 165 kB
stylegan2directions/smile.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:703a67fe070ce1f3c5ad18fc0f5711093756ba1a7048e2e09b2cd6c933032b6e
3
+ size 73856
stylegan2directions/yaw.jpg ADDED

Git LFS Details

  • SHA256: d8559229975b20bb68b2e4d483740907d5d664a5e95a27a3d329a2227812274c
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
stylegan2directions/yaw.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:600bbadd2569a8382f444498bf995b3484b9151972601ddfacd80dfa4d5740cd
3
+ size 73856