在运行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详解请参考。