apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

pad incomplete mask tiles

This commit is contained in:
Sean Sube 2023-04-29 15:00:59 -05:00
parent a4d197a6e7
commit a1e7a5d077
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -13,7 +13,7 @@ from ..params import Border, ImageParams, Size, SizeChart, StageParams
from ..server import ServerContext
from ..utils import is_debug
from ..worker import ProgressCallback, WorkerContext
from .utils import process_tile_grid, process_tile_order
from .utils import process_tile_grid, process_tile_order, complete_tile
logger = getLogger(__name__)
@ -67,6 +67,7 @@ def upscale_outpaint(
left, top, tile = dims
size = Size(*tile_source.size)
tile_mask = stage_mask.crop((left, top, left + tile, top + tile))
tile_mask = complete_tile(tile_mask, tile)
if is_debug():
save_image(server, "tile-source.png", tile_source)