chauhuynh90 commited on
Commit
607c3b0
·
1 Parent(s): d002b83

Update utils/facial_makeup.py

Browse files
Files changed (1) hide show
  1. utils/facial_makeup.py +4 -1
utils/facial_makeup.py CHANGED
@@ -40,7 +40,10 @@ def hair(image, parsing, part=17, color=[230, 50, 20]):
40
  image_hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
41
  tar_hsv = cv2.cvtColor(tar_color, cv2.COLOR_BGR2HSV)
42
 
43
- image_hsv[parsing!=0] = tar_hsv[parsing!=0]
 
 
 
44
 
45
  changed = cv2.cvtColor(image_hsv, cv2.COLOR_HSV2BGR)
46
 
 
40
  image_hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
41
  tar_hsv = cv2.cvtColor(tar_color, cv2.COLOR_BGR2HSV)
42
 
43
+ if part == 12 or part == 13:
44
+ image_hsv[:, :, 0:2] = tar_hsv[:, :, 0:2]
45
+ else:
46
+ image_hsv[:, :, 0:2] = tar_hsv[:, :, 0:2]
47
 
48
  changed = cv2.cvtColor(image_hsv, cv2.COLOR_HSV2BGR)
49