Spaces:
Sleeping
Sleeping
VascoDVRodrigues
commited on
Commit
·
76eeea1
1
Parent(s):
b584f0d
small teak
Browse files- mot-metrics.py +1 -1
mot-metrics.py
CHANGED
@@ -173,7 +173,7 @@ def calculate(predictions, references, max_iou: float = 0.5):
|
|
173 |
raise ValueError("The frame number in the references should be a positive integer")
|
174 |
|
175 |
|
176 |
-
num_frames = max(np_references[:, 0].max(), np_predictions[:, 0].max())
|
177 |
|
178 |
acc = mm.MOTAccumulator(auto_id=True)
|
179 |
for i in range(1, num_frames+1):
|
|
|
173 |
raise ValueError("The frame number in the references should be a positive integer")
|
174 |
|
175 |
|
176 |
+
num_frames = int(max(np_references[:, 0].max(), np_predictions[:, 0].max()))
|
177 |
|
178 |
acc = mm.MOTAccumulator(auto_id=True)
|
179 |
for i in range(1, num_frames+1):
|