Update mota.py

Aspirinkb · Jan 6, 2022 · d742a33 · d742a33
1 parent 8b8d977
commit d742a33
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/mota.py b/tools/mota.py
@@ -51,7 +51,7 @@ def compare_dataframes(gts, ts):
 logger.info('Loading files.')
 
 gt = OrderedDict([(Path(f).parts[-3], mm.io.loadtxt(f, fmt='mot15-2D', min_confidence=1)) for f in gtfiles])
-ts = OrderedDict([(os.path.splitext(Path(f).parts[-1])[0], mm.io.loadtxt(f, fmt='mot15-2D', min_confidence=0.6)) for f in tsfiles])    
+ts = OrderedDict([(os.path.splitext(Path(f).parts[-1])[0], mm.io.loadtxt(f, fmt='mot15-2D', min_confidence=-1.0)) for f in tsfiles])    
 
 mh = mm.metrics.create()    
 accs, names = compare_dataframes(gt, ts)
@@ -81,4 +81,4 @@ def compare_dataframes(gts, ts):
 metrics = mm.metrics.motchallenge_metrics + ['num_objects']
 summary = mh.compute_many(accs, names=names, metrics=metrics, generate_overall=True)
 print(mm.io.render_summary(summary, formatters=mh.formatters, namemap=mm.io.motchallenge_metric_names))
-logger.info('Completed')
+logger.info('Completed')