Recrosoftware/sd-webui-reactor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

save all images in extras tab

Browse Source
This commit is contained in:
Andray 2024-03-21 13:28:01 +04:00
parent ae57149962
commit b7b094ba78
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/reactor_faceswap.py
View File

@ -500,7 +500,7 @@ class FaceSwapScriptExtras(scripts_postprocessing.ScriptPostprocessing):
# TAB MAIN # TAB MAIN
msgs: dict = { msgs: dict = {
"extra_multiple_source": " | Сomparison grid as a result", "extra_multiple_source": "",
} }
img, imgs, select_source, face_model, source_folder, save_original, mask_face, source_faces_index, gender_source, faces_index, gender_target, face_restorer_name, face_restorer_visibility, codeformer_weight, swap_in_source, swap_in_generated, random_image = ui_main.show(is_img2img=False, show_br=False, **msgs) img, imgs, select_source, face_model, source_folder, save_original, mask_face, source_faces_index, gender_source, faces_index, gender_target, face_restorer_name, face_restorer_visibility, codeformer_weight, swap_in_source, swap_in_generated, random_image = ui_main.show(is_img2img=False, show_br=False, **msgs)
@ -684,9 +684,13 @@ class FaceSwapScriptExtras(scripts_postprocessing.ScriptPostprocessing):
if len(result) > 0 and swapped > 0: if len(result) > 0 and swapped > 0:
image = result[0] image = result[0]
if len(result) > 1: if len(result) > 1:
grid = make_grid(result) if hasattr(pp, 'extra_images'):
result.insert(0, grid) image = result[0]
image = grid pp.extra_images.extend(result[1:])
else:
grid = make_grid(result)
result.insert(0, grid)
image = grid
pp.info["ReActor"] = True pp.info["ReActor"] = True
pp.image = image pp.image = image
logger.status("---Done!---") logger.status("---Done!---")