apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use result size helper

This commit is contained in:
Sean Sube 2024-01-05 21:16:10 -06:00
parent e958eff3a8
commit b5ed1101b4
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -160,16 +160,11 @@ class ChainPipeline:
kwargs.pop("params")
# the stage must be split and tiled if any image is larger than the selected/max tile size
must_tile = has_mask(stage_kwargs) or any(
[
needs_tile(
stage_pipe.max_tile,
stage_params.tile_size,
size=kwargs.get("size", None),
source=source,
)
for source in stage_sources.as_images()
]
must_tile = has_mask(stage_kwargs) or needs_tile(
stage_pipe.max_tile,
stage_params.tile_size,
size=kwargs.get("size", None),
source=stage_sources.size(),
)
tile = stage_params.tile_size