diff --git a/scripts/comparison_plots.py b/scripts/comparison_plots.py index a338f97c..a1d9c32c 100644 --- a/scripts/comparison_plots.py +++ b/scripts/comparison_plots.py @@ -16,5 +16,12 @@ data_fol = os.path.join(tracker_folder, dataset) trackers = os.listdir(data_fol) out_loc = os.path.join(plots_folder, dataset) + +if len(sys.argv[1:]) > 0: + if not set(sys.argv[1:]).issubset(set(trackers)): + not_found_trackers = set(sys.argv[1:]) - set(trackers) + raise Exception(f"The following trackers could not be found in {data_fol}: {', '.join(not_found_trackers)}") + trackers = sys.argv[1:] + for cls in classes: trackeval.plotting.plot_compare_trackers(data_fol, trackers, cls, out_loc)