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 |
|