cringe

mot-metrics.py

@@ -189,12 +189,23 @@ def calculate(predictions, references, max_iou: float = 0.5):
     
     return summary
 
-def parse_standard_payload(payload: dict):
+def calculate_from_payload(payload: dict, max_iou: float = 0.5):
     gt_field_name = payload['gt_field_name']
     models = payload['models']
     sequence_list = payload['sequence_list']
 
+    output = {}
+
     for sequence in sequence_list:
+        output[sequence] = {}
+        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])  
+
         for model in models:
             frames = payload['sequences'][sequence][model]
             formated_predictions = []
@@ -205,16 +216,10 @@ def parse_standard_payload(payload: dict):
                     confidence = detection['confidence']
                     confidence = 1  #TODO: remove this line
                     formated_predictions.append([frame_id, id, x, y, w, h, confidence])
+            output[sequence][model] = calculate(formated_predictions, formatted_references, max_iou=max_iou)
+    return output
+            
 
-        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))