github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added transparent mask to draw_landmarks

Browse source
This commit is contained in:
iperov 2019-03-14 12:16:21 +04:00
parent 5bd3f875fa
commit 9823421a44
4 changed files with 18 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
mainscripts/Util.py
View file

@ -63,7 +63,7 @@ def add_landmarks_debug_images(input_path):
if img is not None:
face_landmarks = dflimg.get_landmarks()
LandmarksProcessor.draw_landmarks(img, face_landmarks)
img = LandmarksProcessor.draw_landmarks(img, face_landmarks, transparent_mask=True)
output_file = '{}{}'.format( str(Path(str(input_path)) / filepath.stem), '_debug.jpg')
cv2_imwrite(output_file, img, [int(cv2.IMWRITE_JPEG_QUALITY), 50] )