Use wrapper func to skip empty tiles

nadeemlab · Jul 21, 2023 · dfa1ab8 · dfa1ab8
1 parent ef891fa
commit dfa1ab8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepliif/models/__init__.py b/deepliif/models/__init__.py
@@ -323,7 +323,7 @@ def inference(img, tile_size, overlap_size, model_path, use_torchserve=False, ea
     for i in range(cols):
         for j in range(rows):
             tile = extract_tile(rescaled, tile_size, overlap_size, i, j)
-            res = run_fn(tile, model_path, eager_mode)
+            res = run_wrapper(tile, run_fn, model_path, eager_mode)
 
             stitch_tile(images['Hema'], res['G1'], tile_size, overlap_size, i, j)
             stitch_tile(images['DAPI'], res['G2'], tile_size, overlap_size, i, j)