From 8294feca247c084b5aa1fbb2226ef2c706f6c440 Mon Sep 17 00:00:00 2001 From: iperov Date: Sat, 11 May 2019 17:36:22 +0400 Subject: [PATCH] extractior: fix extract from already extracted dflimg --- mainscripts/Extractor.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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) )