github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix for Issue #13

Browse source 
Swapped + operator for np.concatenate to ensure that the resulting array is 2 by X.
This ensures that we are able to draw the point in the array.
This commit is contained in:
David 2018-08-19 17:41:57 -04:00
parent 61e05f7fdf
commit b877367260
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
facelib/LandmarksProcessor.py
View file

@ -192,7 +192,7 @@ def draw_landmarks (image, image_landmarks, color):
cv2.polylines(image, tuple(np.array([v]) for v in (right_eye, left_eye, mouth)),
True, color, lineType=cv2.LINE_AA)
# the rest of the cicles
for x, y in right_eyebrow+left_eyebrow+mouth+right_eye+left_eye+nose:
for x, y in np.concatenate((right_eyebrow, left_eyebrow, mouth, right_eye, left_eye, nose), axis=0):
cv2.circle(image, (x, y), 1, color, 1, lineType=cv2.LINE_AA)
# jaw big circles
for x, y in jaw: