Do not use marker image in postprocess classification by default

nadeemlab · Aug 12, 2024 · 8088de5 · 8088de5
1 parent e6ff602
commit 8088de5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepliif/models/__init__.py b/deepliif/models/__init__.py
@@ -546,7 +546,7 @@ def inference(img, tile_size, overlap_size, model_path, use_torchserve=False,
  return results # return result images with default key names (i.e., net names)
 
 
-def postprocess(orig, images, tile_size, model, seg_thresh=150, size_thresh='auto', marker_thresh='auto', size_thresh_upper=None):
+def postprocess(orig, images, tile_size, model, seg_thresh=150, size_thresh='auto', marker_thresh=None, size_thresh_upper=None):
  if model == 'DeepLIIF':
  resolution = '40x' if tile_size > 384 else ('20x' if tile_size > 192 else '10x')
  overlay, refined, scoring = compute_results(np.array(orig), np.array(images['Seg']),