def parse_args():
    parser = argparse.ArgumentParser(description='Detect objects in the live video')
    parser.add_argument('--network', dest='network', type=str, default='vgg16_reduced', help='which network to use')
    parser.add_argument('--epoch', dest='epoch', help='epoch of pretrained model', default=1, type=int)
    parser.add_argument('--prefix', dest='prefix', help='Trained model prefix', default=os.path.join(os.getcwd(), 'model', 'ssd'), type=str)
    parser.add_argument('--thresh', dest='thresh', help='Threshold of confidence level', default=0.43, type=float)
    parser.add_argument('--plot-prob', dest='plot_prob', help='Should probabilities be printed. (1 = Yes, 0 = No)', default=1, type=int)
    parser.add_argument('--nms', dest='nms_thresh', type=float, default=0.45, help='non-maximum suppression threshold')
    parser.add_argument('--mean-r', dest='mean_r', type=float, default=123, help='red mean value')
    parser.add_argument('--mean-g', dest='mean_g', type=float, default=117, help='green mean value')
    parser.add_argument('--mean-b', dest='mean_b', type=float, default=104, help='blue mean value')
    parser.add_argument('--data-shape', dest='data_shape', type=int, default=300, help='set image shape')
    parser.add_argument('--class-names', dest='class_names', type=str, default=class_names, help='string of comma separated names')
    parser.add_argument('--force', dest='force_nms', type=bool, default=True, help='force non-maximum suppression on different class')
    parser.add_argument('--has-gpu', dest='gpu', help='GPU device 1 if present else 0', default=1, type=int)
    parser.add_argument('-src', '--source', dest='video_source', type=int, default=0, help='Device index of the camera.')
    parser.add_argument('-wd', '--width', dest='width', type=int, default=480, help='Width of the frames in the video stream.')
    parser.add_argument('-ht', '--height', dest='height', type=int, default=640, help='Height of the frames in the video stream.')
    args = parser.parse_args()
    return args
def parse_args():
    parser = argparse.ArgumentParser(description='Detect objects in the live video')
    parser.add_argument('--network', dest='network', type=str, default='vgg16_reduced', help='which network to use')
    parser.add_argument('--epoch', dest='epoch', help='epoch of pretrained model', default=1, type=int)
    parser.add_argument('--prefix', dest='prefix', help='Trained model prefix', default=os.path.join(os.getcwd(), 'model', 'ssd'), type=str)
    parser.add_argument('--thresh', dest='thresh', help='Threshold of confidence level', default=0.43, type=float)
    parser.add_argument('--plot-prob', dest='plot_prob', help='Should probabilities be printed. (1 = Yes, 0 = No)', default=1, type=int)
    parser.add_argument('--nms', dest='nms_thresh', type=float, default=0.45, help='non-maximum suppression threshold')
    parser.add_argument('--mean-r', dest='mean_r', type=float, default=123, help='red mean value')
    parser.add_argument('--mean-g', dest='mean_g', type=float, default=117, help='green mean value')
    parser.add_argument('--mean-b', dest='mean_b', type=float, default=104, help='blue mean value')
    parser.add_argument('--data-shape', dest='data_shape', type=int, default=300, help='set image shape')
    parser.add_argument('--class-names', dest='class_names', type=str, default=class_names, help='string of comma separated names')
    parser.add_argument('--force', dest='force_nms', type=bool, default=True, help='force non-maximum suppression on different class')
    parser.add_argument('--has-gpu', dest='gpu', help='GPU device 1 if present else 0', default=1, type=int)
    parser.add_argument('-src', '--source', dest='video_source', type=int, default=0, help='Device index of the camera.')
    parser.add_argument('-wd', '--width', dest='width', type=int, default=480, help='Width of the frames in the video stream.')
    parser.add_argument('-ht', '--height', dest='height', type=int, default=640, help='Height of the frames in the video stream.')
    args = parser.parse_args()
    return args