argparse 是关于命令行解析的 Python 标准模块,该模块能让我们直接使用命令行方式向程序中传入参数并让程序运行。
argparse 是 Python 内置的一个用于命令项选项与参数解析的模块,通过在程序中定义好我们需要的参数,argparse 将会从 sys.argv 中解析出这些参数,并自动生成帮助和使用信息。
使用 argparse 的基本流程:
1. import argparse # 首先导入模块
2. parser = argparse.ArgumentParser() # 然后创建一个解析对象(实例化一个 ArgumentParser 类对象)
3. parser.add_argument() # 再向该对象中添加你要关注的命令行参数和选项(调用 add_argument 函数进行命令参数的添加)
4. args = parser.parse_args() # 最后进行解析(调用 parse_args 函数返回参数解析后的对象)
使用 argparse 的实例演示:
import argparse
parser = argparse.ArgumentParser(description='Test Change Detection Models')
parser.add_argument('--model_dir', default='F:/Change_Detection/resnet-50.pth', type='str', help='The path where the model is stored')
parser.add_argument('--train_dataset', default='F:/Change_Detection/train', type='str', help='The path where the training images are stored')
args = parser.parse_args()
# 执行完上述步骤后,即可在程序中使用参数解析后对象调用各命令参数
model_path = args.model_dir # 'F:/Change_Detection/resnet-50.pth'
train_path = args.train_dataset # 'F:/Change_Detection/train'
type 是要传入参数的数据类型,help 是该参数的提示信息。