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

when using unscaled images, don't make image larger than original

Browse source 
...otherwise, if there are "bad landmarks", images could be VERY large
This commit is contained in:
Paul Molodowitch 2019-09-12 15:45:51 -07:00
commit 343f9500c2
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
mainscripts/Extractor.py
View file

@ -229,6 +229,10 @@ class ExtractSubprocessor(Subprocessor):
face_image_size = self.image_size face_image_size = self.image_size
if face_image_size == 0 and self.face_type != FaceType.MARK_ONLY: if face_image_size == 0 and self.face_type != FaceType.MARK_ONLY:
face_image_size = LandmarksProcessor.calc_image_size_for_unscaled(image_landmarks, self.face_type) face_image_size = LandmarksProcessor.calc_image_size_for_unscaled(image_landmarks, self.face_type)
# if landmarks are bad, could get HUGE scale...
# ...so set a max size, same as input image biggest
# dimension
face_image_size = min(face_image_size, max(w,h))
if self.face_type == FaceType.MARK_ONLY: if self.face_type == FaceType.MARK_ONLY:
image_to_face_mat = None image_to_face_mat = None