Spaces:
Sleeping
Sleeping
Siyun He
commited on
Commit
·
59fe7ab
1
Parent(s):
5ed34fb
debug the RGB color
Browse files- .DS_Store +0 -0
- app.py +12 -3
- glasses/.DS_Store +0 -0
.DS_Store
CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
|
|
app.py
CHANGED
@@ -16,7 +16,13 @@ face_mesh = mp_face_mesh.FaceMesh(static_image_mode=False, max_num_faces=1, min_
|
|
16 |
|
17 |
# Initialize the glass number
|
18 |
num = 1
|
19 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
# Count glasses files
|
22 |
def count_files_in_directory(directory):
|
@@ -56,7 +62,10 @@ def change_glasses():
|
|
56 |
num += 1
|
57 |
if num > total_glass_num:
|
58 |
num = 1
|
59 |
-
|
|
|
|
|
|
|
60 |
return overlay
|
61 |
|
62 |
def change_lip_color(frame, color_name='none'):
|
@@ -330,4 +339,4 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple", secondary_hue="blue"))
|
|
330 |
save_button.click(save_frame, [input_img], [])
|
331 |
|
332 |
if __name__ == "__main__":
|
333 |
-
demo.launch(share=True)
|
|
|
16 |
|
17 |
# Initialize the glass number
|
18 |
num = 1
|
19 |
+
overlay_bgr = cv2.imread(f'glasses/glass{num}.png', cv2.IMREAD_UNCHANGED)
|
20 |
+
# Split the channels
|
21 |
+
b, g, r, a = cv2.split(overlay_bgr)
|
22 |
+
# Merge back in RGB order
|
23 |
+
overlay_rgb = cv2.merge((r, g, b, a))
|
24 |
+
# Use overlay_rgb in your process_frame function
|
25 |
+
overlay = overlay_rgb
|
26 |
|
27 |
# Count glasses files
|
28 |
def count_files_in_directory(directory):
|
|
|
62 |
num += 1
|
63 |
if num > total_glass_num:
|
64 |
num = 1
|
65 |
+
overlay_bgr = cv2.imread(f'glasses/glass{num}.png', cv2.IMREAD_UNCHANGED)
|
66 |
+
b, g, r, a = cv2.split(overlay_bgr)
|
67 |
+
overlay_rgb = cv2.merge((r, g, b, a))
|
68 |
+
overlay = overlay_rgb
|
69 |
return overlay
|
70 |
|
71 |
def change_lip_color(frame, color_name='none'):
|
|
|
339 |
save_button.click(save_frame, [input_img], [])
|
340 |
|
341 |
if __name__ == "__main__":
|
342 |
+
demo.launch(share=True)
|
glasses/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|