apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

better mask shifting

This commit is contained in:
HoopyFreud 2023-07-10 22:21:10 -04:00
parent 9b630f97ee
commit 58b5d18608
1 changed files with 16 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
api/onnx_web/diffusers/run.py
View File

@ -244,6 +244,11 @@ def run_inpaint_pipeline(
mask_filter=mask_filter, mask_filter=mask_filter,
) )
if is_debug():
save_image(server, "full-source.png", source)
save_image(server, "full-mask.png", mask)
save_image(server, "full-noise.png", noise)
# check if we can do full-res inpainting if no outpainting is done # check if we can do full-res inpainting if no outpainting is done
logger.debug("border zero: %s", border.isZero()) logger.debug("border zero: %s", border.isZero())
if border.isZero(): if border.isZero():
@ -262,34 +267,17 @@ def run_inpaint_pipeline(
int(mask_center_y + adj_mask_size / 2), int(mask_center_y + adj_mask_size / 2),
) )
if adj_mask_border[0] < 0: # we would like to subtract the excess width (subtract a positive) and add the deficient width (subtract a negative)
adj_mask_border = ( x_adj = -max(adj_mask_border[2] - source.width, 0) - min(adj_mask_border[0], 0)
0, # we would like to subtract the excess height (subtract a negative) and add the deficient height (subtract a negative)
adj_mask_border[1], y_adj = -max(adj_mask_border[3] - source.height, 0) - min(adj_mask_border[1], 0)
adj_mask_border[2] - adj_mask_border[0],
adj_mask_border[3], adj_mask_border = (
) adj_mask_border[0] + x_adj,
if adj_mask_border[1] < 0: adj_mask_border[1] + y_adj,
adj_mask_border = ( adj_mask_border[2] + x_adj,
adj_mask_border[0], adj_mask_border[3] + y_adj,
0, )
adj_mask_border[2],
adj_mask_border[3] - adj_mask_border[1],
)
if adj_mask_border[2] > source.width:
adj_mask_border = (
adj_mask_border[0] - (adj_mask_border[2] - source.width),
adj_mask_border[1],
source.width,
adj_mask_border[3],
)
if adj_mask_border[3] > source.height:
adj_mask_border = (
adj_mask_border[0],
adj_mask_border[1] - (adj_mask_border[3] - source.height),
adj_mask_border[2],
source.height,
)
border_integrity = all( border_integrity = all(
( (
@ -315,11 +303,6 @@ def run_inpaint_pipeline(
save_image(server, "adjusted-mask.png", mask) save_image(server, "adjusted-mask.png", mask)
save_image(server, "adjusted-source.png", source) save_image(server, "adjusted-source.png", source)
if is_debug():
save_image(server, "full-source.png", source)
save_image(server, "full-mask.png", mask)
save_image(server, "full-noise.png", noise)
# set up the chain pipeline and base stage # set up the chain pipeline and base stage
chain = ChainPipeline() chain = ChainPipeline()
stage = StageParams(tile_order=tile_order, tile_size=tile_size) stage = StageParams(tile_order=tile_order, tile_size=tile_size)