test

mot-metrics.py

@@ -188,3 +188,34 @@ def calculate(predictions, references, max_iou: float = 0.5):
         summary[key] = summary[key][0]
     
     return summary
+
+def parse_standard_payload(payload: dict):
+    gt_field_name = payload['gt_field_name']
+    models = payload['models']
+    sequence_list = payload['sequence_list']
+
+    for sequence in sequence_list:
+        for model in models:
+            frames = payload['sequences'][sequence][model]
+            formated_predictions = []
+            for frame_id, frame in enumerate(frames):
+                for detection in frame:
+                    id = detection['index']
+                    x, y, w, h = detection['bounding_box']
+                    confidence = detection['confidence']
+                    confidence = 1  #TODO: remove this line
+                    formated_predictions.append([frame_id, id, x, y, w, h, confidence])
+
+        frames = payload['sequences'][sequence][gt_field_name]
+        formatted_references = []
+        for frame_id, frame in enumerate(frames):
+            for detection in frame:
+                id = detection['index']
+                x, y, w, h = detection['bounding_box']
+                formatted_references.append([frame_id, id, x, y, w, h])    
+
+    print(calculate(formated_predictions, formatted_references, max_iou=0.5))     
+
+
+
+