Spaces:
Sleeping
Sleeping
VascoDVRodrigues
commited on
Commit
·
f8639a1
1
Parent(s):
63e748f
small fix
Browse files- 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 |
-
|
|
|
|
|
|
|
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 |
|