solved a bug and updated show_info statements to inform user about tiling updates

Author: lufre1

synapse_net/inference/util.py

@@ -432,24 +432,24 @@ def get_default_tiling(is_2d: bool = False) -> Dict[str, Dict[str, int]]:
         else:
             raise NotImplementedError(f"Infererence with a GPU with {vram} GB VRAM is not supported.")
 
-        print(f"Determined tile size for CUDA: {tile}")
         tiling = {"tile": tile, "halo": halo}
+        print(f"Determined tile size for CUDA: {tiling}")
 
     elif torch.backends.mps.is_available():  # Check for Apple Silicon (MPS)
         tile = {"x": 512, "y": 512, "z": 64}
         halo = {"x": 64, "y": 64, "z": 16}
-        print(f"Determined tile size for MPS: {tile}")
         tiling = {"tile": tile, "halo": halo}
+        print(f"Determined tile size for MPS: {tiling}")
 
 
     # I am not sure what is reasonable on a cpu. For now choosing very small tiling.
     # (This will not work well on a CPU in any case.)
     else:
-        print(f"Determining default tiling for CPU: {tile}")
         tiling = {
             "tile": {"x": 96, "y": 96, "z": 16},
             "halo": {"x": 16, "y": 16, "z": 4},
         }
+        print(f"Determining default tiling for CPU: {tiling}")
 
     return tiling
 

synapse_net/tools/segmentation_widget.py

@@ -45,7 +45,10 @@ def _get_current_tiling(tiling: dict, default_tiling: dict, model_type: str):
         for k2, v2 in v.items():
             if isinstance(v2, int):
                 continue
-            tiling[k][k2] = v2.value()
+            elif hasattr(v2, "value"):  # If it's a QSpinBox, extract the value
+                tiling[k][k2] = v2.value()
+            else:
+                raise TypeError(f"Unexpected type for tiling value: {type(v2)} at {k}/{k2}")
     # check if user inputs tiling/halo or not
     if default_tiling == tiling:
         if "2d" in model_type:
@@ -54,11 +57,13 @@ def _get_current_tiling(tiling: dict, default_tiling: dict, model_type: str):
                 "tile": {"x": 512, "y": 512, "z": 1},
                 "halo": {"x": 64, "y": 64, "z": 1},
             }
-    elif "2d" in model_type:
+    else:
+        show_info(f"Using custom tiling: {tiling}")
+    if "2d" in model_type:
         # if its a 2d model set z to 1
         tiling["tile"]["z"] = 1
-        tiling["halo"]["z"] = 1
-
+        tiling["halo"]["z"] = 0
+        show_info(f"Using tiling: {tiling}")
     return tiling