VascoDVRodrigues commited on
Commit
e318a9c
1 Parent(s): f8639a1
Files changed (1) hide show
  1. mot-metrics.py +2 -0
mot-metrics.py CHANGED
@@ -212,6 +212,7 @@ def calculate_from_payload(payload: dict, max_iou: float = 0.5, debug: bool = Fa
212
  for model in models:
213
  frames = payload['sequences'][sequence][model]
214
  formated_predictions = []
 
215
  for frame_id, frame in enumerate(frames):
216
  for detection in frame:
217
  id = detection['index']
@@ -220,6 +221,7 @@ def calculate_from_payload(payload: dict, max_iou: float = 0.5, debug: bool = Fa
220
  confidence = 1 #TODO: remove this line
221
  formated_predictions.append([frame_id+1, id, x, y, w, h, confidence])
222
  if debug:
 
223
  print("formated_predictions: ", formated_predictions)
224
  print("formatted_references: ", formatted_references)
225
  if len(formated_predictions) == 0:
 
212
  for model in models:
213
  frames = payload['sequences'][sequence][model]
214
  formated_predictions = []
215
+
216
  for frame_id, frame in enumerate(frames):
217
  for detection in frame:
218
  id = detection['index']
 
221
  confidence = 1 #TODO: remove this line
222
  formated_predictions.append([frame_id+1, id, x, y, w, h, confidence])
223
  if debug:
224
+ print("sequence/model: ", sequence, model)
225
  print("formated_predictions: ", formated_predictions)
226
  print("formatted_references: ", formatted_references)
227
  if len(formated_predictions) == 0: