VascoDVRodrigues commited on
Commit
76eeea1
1 Parent(s): b584f0d

small teak

Browse files
Files changed (1) hide show
  1. 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):