updated baseeval to tolerate a missing iou_score_field

CosmiQ · Oct 11, 2018 · 13509bc · 13509bc
1 parent b140b4e
commit 13509bc
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/cw_eval/baseeval.py b/cw_eval/baseeval.py
@@ -86,8 +86,14 @@ def eval_iou(self, miniou=0.5, iou_field_prefix='iou_score',
                 TruePos = 0
                 FalsePos = 0
             else:
-                TruePos = self.proposal_GDF[self.proposal_GDF[iou_field]>=miniou].shape[0]
-                FalsePos = self.proposal_GDF[self.proposal_GDF[iou_field]<miniou].shape[0]
+                try:
+                    TruePos = self.proposal_GDF[self.proposal_GDF[iou_field]>=miniou].shape[0]
+                    FalsePos = self.proposal_GDF[self.proposal_GDF[iou_field]<miniou].shape[0]
+                except:
+                    print("iou field {} missing")
+                    TruePos = 0
+                    FalsePos = 0
+
 
             FalseNeg = self.ground_truth_GDF_Edit.shape[0]