loop_gt2ds.py

#recusively inference on groundtruth of detection
import os
from tqdm import tqdm
import argparse

def get_parser():
    parser = argparse.ArgumentParser(description="inference vids")
    parser.add_argument(
        "--root_path",
        type=str,
        default='/media/data3/EgoCentric_Nafosted/micand30/gt/',
        help="Root path",
    )
    parser.add_argument(
        "--tracker",
        type=str,
        default="sort",
        help="sort or deepsort",
    )
    parser.add_argument(
        "--out_vids",
        type=str,
        default='/media/data3/EgoCentric_Nafosted/micand30/out_vids/gt2s/',
        help='Path to folder to save output videos',
    )
    parser.add_argument(
        "--out_txts",
        type=str,
        default='/media/data3/EgoCentric_Nafosted/micand30/test/gt2s/',
        help='Path to folder to save output sequence texts',
    )
    return parser

def main():
    args = get_parser().parse_args()
    directories = os.listdir(args.root_path)
    for i in tqdm(range(len(directories))):
        directory = directories[i]
        input_path = os.path.join(args.root_path, directory) 
        out_vid = os.path.join(args.out_vids, directory+'.avi')
        out_txt = os.path.join(args.out_txts, directory+'.txt')
        command = 'python3 gt2ds.py --input ' + input_path + ' --tracker ' + args.tracker + ' --out_vid ' + out_vid + ' --out_txt ' + out_txt
        print(command)
        os.system(command)

    return 0

if __name__ == "__main__":
    main()