parser.add_argument例子

'''名字，type参数类型，default默认值，help:参数描述，metavar'''
def parse_args():
    parser = argparse.ArgumentParser("Evaluation Parameters")
    parser.add_argument( 
        '--image_dir',
        type=str,
        default='./insects/test/images',
        help='the directory of test images')
    parser.add_argument(
        '--weight_file',
        type=str,
        default='./yolo_epoch50',
        help='the path of model parameters')
    args = parser.parse_args()
    return args


args = parse_args()
TESTDIR = args.image_dir#这里,就直接调类的方法调用了,image_dir是上面创建的,创建的时候记得加--
WEIGHT_FILE = args.weight_file