关于运行Python脚本命令行参数传递

六月ゝ 毕业季﹏ 提交于 2019-12-05 20:34:20

在运行Python脚本的时候,需要的参数没有传入的时候就会报错,而又不知道怎么传入。如下参数解析函数就会检查是否按要求传入了所需的参数。如下所示:

def parse_args():
    parser = argparse.ArgumentParser(description='Convert dataset')
    # parser.add_argument(
    #     '--dataset', help="cocostuff, cityscapes", default=None, type=str)
    parser.add_argument(
        '--dataset', help="cocostuff, cityscapes", default=None, type=str)
    parser.add_argument(
        '--outdir', help="output dir for json files", default=None, type=str)
    parser.add_argument(
        '--datadir', help="data dir for annotations to be converted",
        default=None, type=str)
    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(1)
    return parser.parse_args()

如上函数parse_args,就会解析传入的参数并检查参数个数(len(sys.argv) )。注意这里的sys.argv[0]的值就是要运行的脚本文件名,sys.argv[1],...,sys.argv[n]才是要传入的真正参数。所以当len(sys.argv) ==1时就是表示没有传入任何参数。要传入所需的dataset,outdir和datadir参数,可以通过命令行或者通过Python IDE(如PyCharm)等方法。

不论是通过命令行还是Python IDE,设置的方式均是一样的,如

--dataset=MS COCO   --outdir=usr\OutResults   --datadir=usr\data

在PyCharm下传递运行脚本所需要的参数的方式如下:

Edit Configurations -> Configuration -> Script parameters,具体示例如下图所示,把上述参数直接复制粘贴即可。关于python的argparse模块add_argument详解请参考。



易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!