VascoDVRodrigues commited on
Commit
f8639a1
·
1 Parent(s): 63e748f
Files changed (1) hide show
  1. mot-metrics.py +4 -1
mot-metrics.py CHANGED
@@ -222,7 +222,10 @@ def calculate_from_payload(payload: dict, max_iou: float = 0.5, debug: bool = Fa
222
  if debug:
223
  print("formated_predictions: ", formated_predictions)
224
  print("formatted_references: ", formatted_references)
225
- output[sequence][model] = calculate(formated_predictions, formatted_references, max_iou=max_iou)
 
 
 
226
  return output
227
 
228
 
 
222
  if debug:
223
  print("formated_predictions: ", formated_predictions)
224
  print("formatted_references: ", formatted_references)
225
+ if len(formated_predictions) == 0:
226
+ output[sequence][model] = "Model had no predictions."
227
+ else:
228
+ output[sequence][model] = calculate(formated_predictions, formatted_references, max_iou=max_iou)
229
  return output
230
 
231