c

my_metricv2.py

@@ -140,10 +140,15 @@ def calculate(predictions, references, max_iou: float = 0.5):
     for i in range(1, num_frames+1):
         preds = np_predictions[np_predictions[:, 0] == i, 1:6]
         refs = np_references[np_references[:, 0] == i, 1:6]
-        print(preds)
-        print(refs)
+        C = mm.distances.iou_matrix(refs[:,1:], preds[:,1:], max_iou = max_iou)
+        acc.update(refs[:,0].astype('int').tolist(), preds[:,0].astype('int').tolist(), C)
+
+    mh = mm.metrics.create()
+    summary = mh.compute(acc, metrics=['num_frames', 'mota', 'motp'], name='acc')
+
 
     return {
-            "predictions": predictions,
-            "references": references
+            "mota": summary['mota'],
+            "motp": summary['motp'],
+            "num_frames": summary['num_frames']
         }