bascobasculino commited on
Commit
3b2900b
·
1 Parent(s): f2d45cf
Files changed (1) hide show
  1. my_metricv2.py +2 -6
my_metricv2.py CHANGED
@@ -144,11 +144,7 @@ def calculate(predictions, references, max_iou: float = 0.5):
144
  acc.update(refs[:,0].astype('int').tolist(), preds[:,0].astype('int').tolist(), C)
145
 
146
  mh = mm.metrics.create()
147
- summary = mh.compute(acc, metrics=['num_frames', 'mota', 'motp'], name='acc')
148
 
149
 
150
- return {
151
- "mota": summary['mota'],
152
- "motp": summary['motp'],
153
- "num_frames": summary['num_frames']
154
- }
 
144
  acc.update(refs[:,0].astype('int').tolist(), preds[:,0].astype('int').tolist(), C)
145
 
146
  mh = mm.metrics.create()
147
+ summary = mh.compute(acc)
148
 
149
 
150
+ return summary