Merge pull request #118 from shota-0129/fix/missing-return-statement

Fix/missing return statement
wolny · Jul 30, 2024 · 5917023 · 5917023
2 parents 203225f + e2143ec
commit 5917023
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch3dunet/unet3d/config.py b/pytorch3dunet/unet3d/config.py
@@ -49,7 +49,7 @@ def load_config():
     if device == 'cpu':
         logger.warning('CPU mode forced in config, this will likely result in slow training/prediction')
         config['device'] = 'cpu'
-        return config
+        return config, config_path
 
     if torch.cuda.is_available():
         config['device'] = 'cuda'

diff --git a/pytorch3dunet/unet3d/seg_metrics.py b/pytorch3dunet/unet3d/seg_metrics.py
@@ -29,7 +29,7 @@ def _iou_matrix(gt, seg):
     seg = _relabel(seg)
 
     # get number of overlapping pixels between GT and SEG
-    n_inter = contingency_table(gt, seg).A
+    n_inter = contingency_table(gt, seg).toarray()
 
     # number of pixels for GT instances
     n_gt = n_inter.sum(axis=1, keepdims=True)