show grid image

coder168 · Jul 21, 2023 · b3f03b2 · b3f03b2
1 parent 6120830
commit b3f03b2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/use_cases/image_processor.py b/scripts/use_cases/image_processor.py
@@ -64,6 +64,9 @@ def proc_images(self, o: StableDiffusionProcessing, res: Processed, option: Opti
             all_prompts.extend(proc.all_prompts)
             infotexts.extend(proc.infotexts)
 
+        if res.index_of_first_image == 1:
+            edited_images.insert(0, images.image_grid(edited_images))
+
         if option.show_original_image:
             res.images.extend(edited_images)
         else:
@@ -279,7 +282,7 @@ def __show_detected_faces(self, entire_image: np.ndarray, faces: List[Face], p:
                 face.top : face.bottom,
                 face.left : face.right,
             ] = face_image
-        return self.__add_comment(entire_image, f"{len(faces)}")
+        return Image.fromarray(self.__add_comment(entire_image, f"{len(faces)}"))
 
     def __get_wildcards_script(self, p: StableDiffusionProcessingImg2Img):
         if p.scripts is None: