通过命令行向python文件传入参数(argparse.ArgumentParser()模块使用)

import argparse

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(
        "--digit",
        type=int,
        default=123,
        help="输入数字")
    args = parser.parse_args()
    return args


args = parse_args()


print(args.image_dir)
print("..................")
print(args.digit)

使用 argparse 的第一步是创建一个 ArgumentParser 对象,上面定义的parse_args()方法内调用了argparse.ArgumentParser模块如下

parser = argparse.ArgumentParser("Evaluation Parameters"),

第二步添加该模块下参数变量的名字和默认值 ,如下

parser.add_argument(
        '--image_dir',
        type=str,
        default='./insects/test/images',
        help='the directory of test images')

这时直接运行就是打印默认值,如下

然后我们使用命令行测试,输入新的参数 注意下面格式(话说记得,进入文件所在目录哈,不要太哈了)

python cs.py --digit 456

结果如下,可以看见已经改变了,默认的数字

总结!一定记得前面那两根 – 啊!!!