()
return args
Parse input arguments
Parse input arguments
Parse input arguments
def parse_args():
    """
    Parse input arguments
    """
    parser = argparse.ArgumentParser(description='TestImgs...U need provide the test dir')
    parser.add_argument('--data_dir', dest='data_dir',
                        help='data path',
                        default='demos', type=str)
    parser.add_argument('--save_dir', dest='save_dir',
                        help='demo imgs to save',
                        default='inference_results', type=str)
    parser.add_argument('--gpu', dest='gpu',
                        help='gpu id ',
                        default='0', type=str)

    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(1)

    args = parser.parse_args()

    return args
def parse_args ( ) : parser = argparse . ArgumentParser ( description = 'TestImgs...U need provide the test dir' ) parser . add_argument ( '--data_dir' , dest = 'data_dir' , help = 'data path' , default = 'demos' , type = str ) parser . add_argument ( '--save_dir' , dest = 'save_dir' , help = 'demo imgs to save' , default = 'inference_results' , type = str ) parser . add_argument ( '--gpu' , dest = 'gpu' , help = 'gpu id ' , default = '0' , type = str ) if len ( sys . argv ) == 1 : parser . print_help ( ) sys . exit ( 1 ) args = parser . parse_args ( ) return args