fix(api): log tile size and overlap

ssube · Jul 2, 2023 · eef055e · eef055e
1 parent 4b84188
commit eef055e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/api/onnx_web/chain/stage.py b/api/onnx_web/chain/stage.py
@@ -2,9 +2,9 @@
 
 from PIL import Image
 
-from onnx_web.params import ImageParams, Size, SizeChart, StageParams
-from onnx_web.server.context import ServerContext
-from onnx_web.worker.context import WorkerContext
+from ..params import ImageParams, Size, SizeChart, StageParams
+from ..server.context import ServerContext
+from ..worker.context import WorkerContext
 
 
 class BaseStage:

diff --git a/api/onnx_web/chain/utils.py b/api/onnx_web/chain/utils.py
@@ -147,6 +147,14 @@ def process_tile_grid(
     tiles_x = ceil(width / adj_tile)
     tiles_y = ceil(height / adj_tile)
     total = tiles_x * tiles_y
+    logger.debug(
+        "processing %s tiles (%s x %s) with adjusted size of %s, %s overlap",
+        total,
+        tiles_x,
+        tiles_y,
+        adj_tile,
+        overlap,
+    )
 
     tiles: List[Tuple[int, int, Image.Image]] = []