diff --git a/mainscripts/Extractor.py b/mainscripts/Extractor.py index e544a0e..bb9873a 100644 --- a/mainscripts/Extractor.py +++ b/mainscripts/Extractor.py @@ -214,6 +214,10 @@ class ExtractSubprocessor(Subprocessor): else: face_idx = 0 for rect, image_landmarks in zip( rects, landmarks ): + if src_dflimg is not None and face_idx > 1: + #cannot extract more than 1 face from dflimg + break + if image_landmarks is None: continue @@ -238,8 +242,8 @@ class ExtractSubprocessor(Subprocessor): if self.debug_dir is not None: LandmarksProcessor.draw_rect_landmarks (debug_image, rect, image_landmarks, self.image_size, self.face_type, transparent_mask=True) - if src_dflimg is not None: - #if extracting from dflimg copy it in order not to lose quality + if src_dflimg is not None and filename_path.suffix == '.jpg': + #if extracting from dflimg and jpg copy it in order not to lose quality output_file = str(self.final_output_path / filename_path.name) if str(filename_path) != str(output_file): shutil.copy ( str(filename_path), str(output_file) )