oskarastrom commited on
Commit
31371fc
·
1 Parent(s): 03cb39e

Update track_eval.py

Browse files
Files changed (1) hide show
  1. scripts/track_eval.py +2 -5
scripts/track_eval.py CHANGED
@@ -37,12 +37,9 @@ def main(args):
37
 
38
  track(infer_args, config=config, verbose=args.verbose)
39
 
40
- result = evaluate(infer_args.output, "../frames/MOT", "../frames/metadata", infer_args.tracker, False)
41
  metrics = result['MotChallenge2DBox']['tracker']['COMBINED_SEQ']['pedestrian']
42
- hota = metrics['HOTA']['HOTA']
43
- print(hota)
44
- mean_hota = np.mean(hota)
45
- print('HOTA:', mean_hota)
46
  print('MOTA:', metrics['CLEAR']['MOTA'])
47
  print('IDF1:', metrics['Identity']['IDF1'])
48
  print('nMAE:', metrics['nMAE']['nMAE'])
 
37
 
38
  track(infer_args, config=config, verbose=args.verbose)
39
 
40
+ result = evaluate(infer_args.output, "../frames/MOT", "../frames/metadata", infer_args.tracker, True)
41
  metrics = result['MotChallenge2DBox']['tracker']['COMBINED_SEQ']['pedestrian']
42
+ print('HOTA:', np.mean(metrics['HOTA']['HOTA']))
 
 
 
43
  print('MOTA:', metrics['CLEAR']['MOTA'])
44
  print('IDF1:', metrics['Identity']['IDF1'])
45
  print('nMAE:', metrics['nMAE']['nMAE'])