argparse

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()解析添加的参数

猜你喜欢

转载自blog.csdn.net/qq_15258623/article/details/80921634