Update track_eval.py

scripts/track_eval.py

@@ -2,6 +2,7 @@ import project_path
 import argparse
 from track_detection import main as track
 import sys
+import numpy as np
 sys.path.append('..')
 sys.path.append('../caltech-fish-counting')
 
@@ -38,7 +39,10 @@ def main(args):
 
     result = evaluate(infer_args.output, "../frames/MOT", "../frames/metadata", infer_args.tracker, False)
     metrics = result['MotChallenge2DBox']['tracker']['COMBINED_SEQ']['pedestrian']
-    print('HOTA:', '?')
+    hota = metrics['HOTA']['HOTA']
+    print(hota)
+    mean_hota = np.mean(hota)
+    print('HOTA:', mean_hota)
     print('MOTA:', metrics['CLEAR']['MOTA'])
     print('IDF1:', metrics['Identity']['IDF1'])
     print('nMAE:', metrics['nMAE']['nMAE'])