openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-15 15:13:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update images.py

Browse Source 
Handles grids names. Code more robust and doesn't fail if mixed with other files.
This commit is contained in:
Michoko 2022-09-13 17:46:05 +02:00 committed by AUTOMATIC1111
parent 94aeb5dec9
commit d5520d43fd
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
modules/images.py
View File

@ -279,17 +279,24 @@ def apply_filename_pattern(x, p, seed, prompt):
return x return x
def get_next_sequence_number(path): def get_next_sequence_number(path, checkAtEnd = False):
""" """
Determines and returns the next sequence number to use when saving an image in the specified directory. Determines and returns the next sequence number to use when saving an image in the specified directory.
set checkAtEnd to True if the sequence is at the end of the filename
The sequence starts at 0. The sequence starts at 0.
""" """
result = -1 result = -1
for p in os.listdir(path): for p in os.listdir(path):
if p.endswith(('.png', '.jpg')): l = os.path.splitext(p)[0].split('-')
result = max(int(p.split('-')[0]), result) try:
if checkAtEnd:
result = max(int(l[-1]), result)
else:
result = max(int(l[0]), result)
except ValueError:
pass
return result + 1 return result + 1
def save_image(image, path, basename, seed=None, prompt=None, extension='png', info=None, short_filename=False, no_prompt=False, grid=False, pnginfo_section_name='parameters', p=None, existing_info=None): def save_image(image, path, basename, seed=None, prompt=None, extension='png', info=None, short_filename=False, no_prompt=False, grid=False, pnginfo_section_name='parameters', p=None, existing_info=None):
@ -324,7 +331,7 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
os.makedirs(path, exist_ok=True) os.makedirs(path, exist_ok=True)
basecount = get_next_sequence_number(path) basecount = get_next_sequence_number(path, basename != '')
fullfn = "a.png" fullfn = "a.png"
fullfn_without_extension = "a" fullfn_without_extension = "a"
for i in range(500): for i in range(500):