apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

linting

This commit is contained in:
HoopyFreud 2023-07-08 21:30:54 -04:00
parent 80e124554d
commit 3fbf9baae6
1 changed files with 16 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
api/onnx_web/chain/tile.py
View File

@ -269,19 +269,18 @@ def process_tile_spiral(
needs_margin = True
bottom_margin = height - bottom
#if no source given, we don't have a source image
# if no source given, we don't have a source image
if not source:
tile_image = None
elif needs_margin:
#in the special case where the image is smaller than the specified tile size, just use the image
# in the special case where the image is smaller than the specified tile size, just use the image
if single_tile:
logger.debug("creating and processing single-tile subtile")
tile_image = source
#otherwise use add histogram noise outside of the image border
# otherwise use add histogram noise outside of the image border
else:
logger.debug("tiling and adding margin")
base_image = (
source.crop(
base_image = source.crop(
(
left + left_margin,
top + top_margin,
@ -289,7 +288,6 @@ def process_tile_spiral(
bottom - bottom_margin,
)
)
)
tile_image = noise_source_histogram(base_image, (tile, tile), (0, 0))
tile_image.paste(base_image, (left_margin, top_margin))
else:
@ -355,10 +353,7 @@ def generate_tile_spiral(
span_x = tile + (width_tile_target - 1) * tile_increment
span_y = tile + (height_tile_target - 1) * tile_increment
logger.debug(
"tiled image overlap: %s. Span: %s x %s",
overlap,span_x,span_y
)
logger.debug("tiled image overlap: %s. Span: %s x %s", overlap, span_x, span_y)
tile_left = (
width - span_x