apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): calculate outpaint overlap factor before adding margins

This commit is contained in:
Sean Sube 2023-02-06 22:49:58 -06:00
parent a9f4e19616
commit b6fd288842
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/chain/upscale_outpaint.py
View File

@ -35,6 +35,10 @@ def upscale_outpaint(
prompt = prompt or params.prompt
logger.info("upscaling image by expanding borders: %s", border)
margin_x = float(max(border.left, border.right))
margin_y = float(max(border.top, border.bottom))
overlap = min(margin_x / source_image.width, margin_y / source_image.height)
if mask_image is None:
# if no mask was provided, keep the full source image
mask_image = Image.new("RGB", source_image.size, "black")
@ -108,10 +112,6 @@ def upscale_outpaint(
draw_mask.rectangle((left, top, left + tile, top + tile), fill="black")
return result.images[0]
margin_x = float(max(border.left, border.right))
margin_y = float(max(border.top, border.bottom))
overlap = min(margin_x / source_image.width, margin_y / source_image.height)
if overlap == 0:
logger.debug("outpainting with 0 margin, using grid tiling")
output = process_tile_grid(source_image, SizeChart.auto, 1, [outpaint])