Spaces:
Sleeping
Sleeping
VascoDVRodrigues
commited on
Commit
·
e318a9c
1
Parent(s):
f8639a1
- mot-metrics.py +2 -0
mot-metrics.py
CHANGED
@@ -212,6 +212,7 @@ def calculate_from_payload(payload: dict, max_iou: float = 0.5, debug: bool = Fa
|
|
212 |
for model in models:
|
213 |
frames = payload['sequences'][sequence][model]
|
214 |
formated_predictions = []
|
|
|
215 |
for frame_id, frame in enumerate(frames):
|
216 |
for detection in frame:
|
217 |
id = detection['index']
|
@@ -220,6 +221,7 @@ def calculate_from_payload(payload: dict, max_iou: float = 0.5, debug: bool = Fa
|
|
220 |
confidence = 1 #TODO: remove this line
|
221 |
formated_predictions.append([frame_id+1, id, x, y, w, h, confidence])
|
222 |
if debug:
|
|
|
223 |
print("formated_predictions: ", formated_predictions)
|
224 |
print("formatted_references: ", formatted_references)
|
225 |
if len(formated_predictions) == 0:
|
|
|
212 |
for model in models:
|
213 |
frames = payload['sequences'][sequence][model]
|
214 |
formated_predictions = []
|
215 |
+
|
216 |
for frame_id, frame in enumerate(frames):
|
217 |
for detection in frame:
|
218 |
id = detection['index']
|
|
|
221 |
confidence = 1 #TODO: remove this line
|
222 |
formated_predictions.append([frame_id+1, id, x, y, w, h, confidence])
|
223 |
if debug:
|
224 |
+
print("sequence/model: ", sequence, model)
|
225 |
print("formated_predictions: ", formated_predictions)
|
226 |
print("formatted_references: ", formatted_references)
|
227 |
if len(formated_predictions) == 0:
|