parser.add_argument(),里面的参数,如果有–,表示是可选参数,没有–的话,意味着是必选参数,在运行时必须输入,default是没有用的。
其它的基本解释可以参考python add_argument()用法解析
def main():
parser = ArgumentParser()
parser.add_argument('--img', default='demo.png', help='Image file')
parser.add_argument('--config', default='..\configs\pspnet\pspnet_r50-d8_512x1024_40k_cityscapes.py',
help='Config file')
parser.add_argument('--checkpoint',
default='..\checkpoints\pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth',
help='Checkpoint file')
parser.add_argument(
'--device', default='cpu', help='Device used for inference')
parser.add_argument(
'--palette',
default='cityscapes',
help='Color palette used for segmentation map')
args = parser.parse_args()