apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use presence of stage mask to trigger tiling

This commit is contained in:
Sean Sube 2023-11-29 21:52:32 -06:00
parent 042181b9c5
commit 48d0d3e22d
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -146,7 +146,7 @@ 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 = any(
must_tile = "mask" in stage_kwargs or any(
[
needs_tile(
stage_pipe.max_tile,
@ -162,7 +162,7 @@ class ChainPipeline:
if stage_pipe.max_tile > 0:
tile = min(stage_pipe.max_tile, stage_params.tile_size)
if must_tile or len(stage_sources) > 0:
if must_tile:
logger.info(
"image contains sources or is larger than tile size of %s, tiling stage",
tile,