developer-mayuan
12/13/2017 - 12:05 AM

Experiment argument parser in Python using argparser

Experiment argument parser in Python using argparser

def parse_args():
    """Parse input arguments."""
    parser = argparse.ArgumentParser(
        description='Head pose estimation using the Hopenet network.')
    parser.add_argument('--gpu', dest='gpu_id',
                        help='GPU device id to use [0]',
                        default=0, type=int)
    parser.add_argument('--snapshot', dest='snapshot',
                        help='Path of model snapshot.',
                        default='', type=str)
    parser.add_argument('--video', dest='video_path', help='Path of video')
    parser.add_argument('--bboxes', dest='bboxes',
                        help='Bounding box annotations of frames')
    parser.add_argument('--output_string', dest='output_string',
                        help='String appended to output file')
    parser.add_argument('--n_frames', dest='n_frames',
                        help='Number of frames', type=int)
    parser.add_argument('--fps', dest='fps',
                        help='Frames per second of source video',
                        type=float, default=30.)
    args = parser.parse_args()
    return args