single_cam_evaluation.py

import argparse

from evaluation.motmetrics_evaluation import eval_single_cam_multiple_cams


def arguement_inputs():
    parser = argparse.ArgumentParser(description='Single cam evaluation')
    parser.add_argument('--test_data_dir', type=str, required=True, help='MTA test data directory')
    parser.add_argument('--tracking_results_dir', type=str, required=True, help='tracking results directory')
    parser.add_argument('--working_dir', type=str, required=True, help='working directory')
    parser.add_argument('--cam_ids', type=int, nargs='+', required=True, help='camera ids for evaluation')
    parser.add_argument('--eval_results_dir', type=str, required=True, help='evaluation results directory')
    return parser.parse_args()

def main():
    args = arguement_inputs()
    eval_single_cam_multiple_cams(dataset_base=args.test_data_dir,
                                  track_results_folder=args.tracking_results_dir,
                                  working_dir=args.working_dir,
                                  cam_ids=args.cam_ids,
                                  results_output_folder=args.eval_results_dir
                                  )

if __name__ == "__main__":
    main()