Spaces:
Runtime error
Runtime error
Commit
·
607c3b0
1
Parent(s):
d002b83
Update utils/facial_makeup.py
Browse files- 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 |
-
|
|
|
|
|
|
|
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 |
|