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

Fix/missing return statement

Author: wolny

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'

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)