tenso

my_metricv2.py

@@ -117,13 +117,17 @@ def calculate(predictions, references):
     """Returns the scores"""
     try: 
         np_predictions = np.array(predictions)
+    except:
+        raise ValueError("The predictions should be a list of np.arrays in the format [frame number, object id, bb_left, bb_top, bb_width, bb_height, confidence]")
+    
+    try:
         np_references = np.array(references)
     except:
-        raise ValueError("The predictions and references should be lists of floats in the correct format [frame number, object id, bb_left, bb_top, bb_width, bb_height, confidence]")
+        raise ValueError("The references should be a list of np.arrays in the format [frame number, object id, bb_left, bb_top, bb_width, bb_height, confidence]")
     
-    print(predictions)
-    print(references)
-
+    num_frames = max(np_references[:, 0].max(), np_predictions[:, 0].max())+1
+        # acc = mm.MOTAccumulator(auto_id=True)
+    print(num_frames)
     return {
         
             "predictions": predictions,