mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix resize

Browse Source
This commit is contained in:
lllyasviel 2023-10-25 21:02:55 -07:00
parent 959f965b77
commit 214c2ba04c
3 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fooocus_version.py
View File

@ -1 +1 @@
version = '2.1.743' version = '2.1.744'

8
modules/async_worker.py
View File

@ -31,7 +31,7 @@ def worker():
from modules.private_logger import log from modules.private_logger import log
from modules.expansion import safe_str from modules.expansion import safe_str
from modules.util import join_prompts, remove_empty_str, HWC3, resize_image, \ from modules.util import join_prompts, remove_empty_str, HWC3, resize_image, \
get_image_shape_ceil, set_image_shape_ceil, get_shape_ceil get_image_shape_ceil, set_image_shape_ceil, get_shape_ceil, resample_image
from modules.upscaler import perform_upscale from modules.upscaler import perform_upscale
try: try:
@ -329,10 +329,14 @@ def worker():
f = 1.0 f = 1.0
shape_ceil = get_shape_ceil(H * f, W * f) shape_ceil = get_shape_ceil(H * f, W * f)
if shape_ceil < 1024: if shape_ceil < 1024:
print(f'[Upscale] Image is resized because it is too small.') print(f'[Upscale] Image is resized because it is too small.')
uov_input_image = set_image_shape_ceil(uov_input_image, 1024)
shape_ceil = 1024 shape_ceil = 1024
uov_input_image = set_image_shape_ceil(uov_input_image, shape_ceil) else:
uov_input_image = resample_image(uov_input_image, width=W * f, height=H * f)
image_is_super_large = shape_ceil > 2800 image_is_super_large = shape_ceil > 2800
if 'fast' in uov_method: if 'fast' in uov_method:

2
modules/util.py
View File

@ -12,7 +12,7 @@ LANCZOS = (Image.Resampling.LANCZOS if hasattr(Image, 'Resampling') else Image.L
def resample_image(im, width, height): def resample_image(im, width, height):
im = Image.fromarray(im) im = Image.fromarray(im)
im = im.resize((width, height), resample=LANCZOS) im = im.resize((int(width), int(height)), resample=LANCZOS)
return np.array(im) return np.array(im)