import argparse def args_parse(): # construct the argument parse and parse the arguments ap = argparse.ArgumentParser() ap.add_argument("-dataset_test", required=True, help="path to input dataset_test") ap.add_argument("--dataset_train", required=True, help="path to input dataset_train") ap.add_argument("--model", required=True, help="path to output model") ap.add_argument("--plot", type=str, default="plot.png", help="path to output accuracy/loss plot") args = vars(ap.parse_args())#返回一个字典 键值是参数名称 值是输入的参数 print(args) return args args=args_parse() a=args["dataset_test"] b=args["dataset_train"] print(a)
print(b)
1、创建一个ArgumentParser对象
2、调用add_argument()方法添加 参数,“canshu”必须要有参数“--canshu”可有可无参数“ -canshu”--canshu的别名
requried=TRUE 表示必须要写的参数,type=int 表示参数类型为int形
3、parse_args()解析添加的参数