python argparse
2021-05-19 本文已影响0人
1037号森林里一段干木头
简介:argparse是python内置的一个解析命令行参数的包,这里记录一点简单用法,一般情况下都够用了。
参考链接
import argparse
def main():
#初始化命令行解析器
parser = argparse.ArgumentParser(description="the path processer")
#添加参数
parser.add_argument("-t","--trainDataPath",help="train data path")
parser.add_argument("-v","--valDataPath",help="validation data path")
parser.add_argument("-s","--onnxSavePath",help="the onnx model save path")
#action="store_true" 表示如果有输入 则=True ,default=False表示默认为False
parser.add_argument("-m","--ifSaveModel",action="store_true",default=False,help="if save the pt model")
parser.add_argument("-p","--parserHelp",action="store_true",default=False,help="if show the parser_help")
#指定变量类型
parser.add_argument("-e", "--epoches",type=int ,default=50,help="training epoches")
args = parser.parse_args()
trainDataPath = args.trainDataPath
valDataPath = args.valDataPath
onnxSavePath = args.onnxSavePath
saveFlag = args.ifSaveModel
print("trainDataPath:\t",trainDataPath)
print("valDataPath:\t",valDataPath)
print("onnxSavePath:\t",onnxSavePath)
print("ifSaveModel:\t",saveFlag)
if args.parserHelp:
print("="*100)
parser.print_help()
if __name__ == "__main__":
main()
这个示例很简单了,看一眼应该就知道怎么用了。
image.png