Merge pull request ototadana#118 from ototadana/add-save_original_on_…

…detection_fail-option add save_original_on_detection_fail option
coder168 · Jul 4, 2023 · 341d0b4 · 341d0b4
2 parents 033cdcb + 4001cf2
commit 341d0b4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/scripts/ui/ui_builder.py b/scripts/ui/ui_builder.py
@@ -197,6 +197,11 @@ def on_ui_settings():
         ),
     )
 
+    shared.opts.add_option(
+        "face_editor_save_original_on_detection_fail",
+        shared.OptionInfo(False, "Save original image if face detection fails", gr.Checkbox, section=section),
+    )
+
     shared.opts.add_option(
         "face_editor_script_index",
         shared.OptionInfo(

diff --git a/scripts/use_cases/image_processor.py b/scripts/use_cases/image_processor.py
@@ -113,7 +113,14 @@ def proc_image(
             output_images.append(self.__show_detected_faces(np.copy(entire_image), faces, p))
 
         print(f"number of faces: {len(faces)}")
-        if len(faces) == 0 and pre_proc_image is not None:
+        if (
+            len(faces) == 0
+            and pre_proc_image is not None
+            and (
+                option.save_original_image
+                or not shared.opts.data.get("face_editor_save_original_on_detection_fail", False)
+            )
+        ):
             return Processed(
                 p, images_list=[pre_proc_image], all_prompts=[p.prompt], all_seeds=[p.seed], infotexts=[""]
             )