python之argparse使用

argparse模块常用于手动传入参数。一般的深度学习工程中，存在大量的参数，基本都是采用这个模块进行加载参数，而不是在需要参数的位置填写, 方便进行消融实验。

使用这个模块只需要三个步骤：

1.创建 ArgumentParser() 对象
2.调用 add_argument() 方法添加参数
3.使用 parse_args() 解析添加的参数

示例：

import argparse  # 导入解析模块

def parse_args(script):
    parser = argparse.ArgumentParser(description= 'model script %s' %(script))   # 创建ArgumentParser()对象

    if script == 'train':
        parser.add_argument('--dataset', default= 'cars', type=str, help='choose dataset')  # 调用add_argument()方法添加参数
    if script == 'test':
        parser.add_argument('--model', default='ResNet18', type=str, help='choose model')
    
    return parser.parse_args()  # 使用parse_args()解析添加的参数

params1 = parse_args('train')
params2 = parse_args('test')
print(params1.dataset)  # 获得传入的参数
print(params2.model)

输出：

cars
ResNet18