Siyun He commited on
Commit
59fe7ab
·
1 Parent(s): 5ed34fb

debug the RGB color

Browse files
Files changed (3) hide show
  1. .DS_Store +0 -0
  2. app.py +12 -3
  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
- overlay = cv2.imread(f'glasses/glass{num}.png', cv2.IMREAD_UNCHANGED)
 
 
 
 
 
 
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
- overlay = cv2.imread(f'glasses/glass{num}.png', cv2.IMREAD_UNCHANGED)
 
 
 
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