fix for metric mcov in trainig

SMLC-NYSBC · Apr 28, 2023 · e0e7e6f · e0e7e6f
1 parent 568c783
commit e0e7e6f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/tardis_pytorch/dist_pytorch/trainer.py b/tardis_pytorch/dist_pytorch/trainer.py
@@ -326,7 +326,8 @@ def _validate(self):
                 input0_1 = self.Graph0_25.patch_to_segment(
                     graph=edge_cpu, coord=coord, idx=out_cpu, prune=5, sort=False
                 )
-                mcov0_25.append(mcov(input0_1, target))
+                mcov_m, _ = mcov(input0_1, target)
+                mcov0_25.append(mcov_m)
             except:
                 mcov0_25.append(0.0)
 
@@ -335,7 +336,8 @@ def _validate(self):
                 input0_5 = self.Graph0_5.patch_to_segment(
                     graph=edge_cpu, coord=coord, idx=out_cpu, prune=5, sort=False
                 )
-                mcov0_5.append(mcov(input0_5, target))
+                mcov_m, _ = mcov(input0_5, target)
+                mcov0_5.append(mcov_m)
             except:
                 mcov0_5.append(0.0)
 
@@ -344,7 +346,8 @@ def _validate(self):
                 input0_9 = self.Graph0_9.patch_to_segment(
                     graph=edge_cpu, coord=coord, idx=out_cpu, prune=5, sort=False
                 )
-                mcov0_9.append(mcov(input0_9, target))
+                mcov_m = mcov(input0_9, target)
+                mcov0_9.append(mcov_m)
             except:
                 mcov0_9.append(0.0)
 

diff --git a/tardis_pytorch/utils/metrics.py b/tardis_pytorch/utils/metrics.py
@@ -328,14 +328,12 @@ def mcov(
 
     # Get GT instances, compute IoU for best mach between GT and input
     for j in unique_target:
-        print(j)
         g = targets[targets[:, 0] == j, 1:]  # Pick GT instance
         w_g = g.shape[0] / targets.shape[0]  # ratio of instance to whole PC
         iou = []
 
         # Select max IoU (the best mach)
         for i in unique_input:
-            print(i)
             p = input[input[:, 0] == i, 1:]  # Pick input instance
 
             # Intersection of coordinates between GT and input instances