VascoDVRodrigues commited on
Commit
18cf4f9
1 Parent(s): e318a9c
Files changed (1) hide show
  1. mot-metrics.py +3 -1
mot-metrics.py CHANGED
@@ -223,9 +223,11 @@ def calculate_from_payload(payload: dict, max_iou: float = 0.5, debug: bool = Fa
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:
228
  output[sequence][model] = "Model had no predictions."
 
 
229
  else:
230
  output[sequence][model] = calculate(formated_predictions, formatted_references, max_iou=max_iou)
231
  return output
 
223
  if debug:
224
  print("sequence/model: ", sequence, model)
225
  print("formated_predictions: ", formated_predictions)
226
+ print("formated_references: ", formatted_references)
227
  if len(formated_predictions) == 0:
228
  output[sequence][model] = "Model had no predictions."
229
+ elif len(formatted_references) == 0:
230
+ output[sequence][model] = "No ground truth."
231
  else:
232
  output[sequence][model] = calculate(formated_predictions, formatted_references, max_iou=max_iou)
233
  return output