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

optimizations of nnlib and SampleGeneratorFace,

Browse source 
refactorings
This commit is contained in:
iperov 2019-01-22 11:52:04 +04:00
commit b6c4171ea1
9 changed files with 175 additions and 79 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/ConverterMasked.py
View file

@ -81,7 +81,7 @@ class ConverterMasked(ConverterBase):
img_size = img_bgr.shape[1], img_bgr.shape[0]
img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr, img_face_landmarks)
img_face_mask_a = LandmarksProcessor.get_image_hull_mask (img_bgr.shape, img_face_landmarks)
face_mat = LandmarksProcessor.get_transform_mat (img_face_landmarks, self.output_size, face_type=self.face_type)
face_output_mat = LandmarksProcessor.get_transform_mat (img_face_landmarks, self.output_size, face_type=self.face_type, scale=self.output_face_scale)