apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): convert spiral grid coordinates to offsets

This commit is contained in:
Sean Sube 2023-01-28 23:58:04 -06:00
parent 03d88dff5f
commit 6f1ffaa043
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,7 +51,10 @@ def process_tile_spiral(
width, height = source.size width, height = source.size
image = Image.new('RGB', (width * scale, height * scale)) image = Image.new('RGB', (width * scale, height * scale))
image.paste(source, (0, 0)) image.paste(source, (0, 0, width, height))
center_x = (width / 2) - (tile / 2)
center_y = (height / 2) - (tile / 2)
# TODO: only valid for overlap = 0.5 # TODO: only valid for overlap = 0.5
if overlap == 0.5: if overlap == 0.5:
@ -69,6 +72,9 @@ def process_tile_spiral(
# tile tuples is source, multiply by scale for dest # tile tuples is source, multiply by scale for dest
counter = 0 counter = 0
for left, top in tiles: for left, top in tiles:
left = center_x + int(left)
top = center_y + int(top)
logger.info('processing tile %s of %s', counter, len(tiles)) logger.info('processing tile %s of %s', counter, len(tiles))
counter += 1 counter += 1