parser.add_argument例子

最新推荐文章于 2024-07-17 08:45:22 发布

程序员进化不脱发!

最新推荐文章于 2024-07-17 08:45:22 发布

阅读量849

点赞数

文章标签： python

本文链接：https://blog.csdn.net/weixin_43134049/article/details/105847168

版权

'''名字，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