argparse.ArgumentParser()用法解析
一、简介:
argparse是一个Python模块:命令行选项、参数和子命令解析器,argparse模块可以让人轻松编写用户友好的命令行接口。程序定义它需要的参数。然后argparser将弄清如何从sys.argv解析出那些参数。argparse模块还会自动生成帮助和使用手册,并在用户给程序传入无效参数时报出错误信息。
二、使用流程
1. 创建解析器
parser = argparse.ArgumentParser()
使用argparser的第一步是创建一个ArgumentParser对象,ArgumentParser对象包含将命令行解析成Python数据类型所需的全部信息。
2. 添加参数
parser.add_argument("--epoch", type=int, default=200)#最大训练batch
给一个ArgumentParser添加程序阐述信息是通过调用add_arguement()方法完成的
add_arguement()方法的部分参数解释:
argumentParser.add_argument(name or flags...[, action][, nargs]
[, const][, default][, type][, choices]
[, required][, help][,metavar][,dest])
- name or flags:
未知参数或可选参数的命名或者一个选项字符串的列表
(可选参数是以’-‘为前缀的参数,要经过命令才会输入的;剩下的就是位置参数,位置参数是按照输入的顺序赋值。)
parser.add_argument('-f', '--foo') #指定一个可选参数
parser.add_argument('bar') #指定一个位置参数
action - 表示该选项要执行的操作,默认值是 store
(参考文章:https://blog.csdn.net/foneone/article/details/103984895)
- default - 当参数未在命令行中出现时使用的值
- dest - 用来指定参数的位置
- type - 为参数类型,例如int
- choices - 用来选择输入参数的范围。例如choice = [1, 5, 10], 表示输入参数 只能为1,5 或10
- help - 用来描述这个选项的作用
关于参数更详细的讲解参考文章:https://blog.csdn.net/guoyajie1990/article/details/76739977
3、解析参数
>>>parser.parse_args(['--sum', '7', '-1', '42' ])
或者:
parser.add_argument("--epoch", type=int, default=200)#最大训练batch
args = parser.parse_args()
epoch = args.epoch
ArgumentParser通过parse_args()方法解析参数
4、总代码示例
以下是深度学习训练图片数据时用到的部分参数代码,仅供参考
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--n_classes", type=int, default=2)#数据类别数
parser.add_argument("--input_height", type=int, default=80)#数据高度
parser.add_argument("--input_width", type=int, default=80)#数据宽度
parser.add_argument("--input_train_img", type=str, default="./data_train/")#输入图像地址
parser.add_argument("--lr", type=float, default=0.001)#学习率
args = parser.parse_args()
n_classes = args.n_classes
input_height = args.input_height
input_width = args.input_width
input_train_img = args.input_train_img
lr = args.lr