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
上一篇下一篇

猜你喜欢

热点阅读