openmmlab(三)图像分类配置文件&训练&推理

Get start with Python API

    在有了训练配置文件和训练权重的前提下，可以用此API进行快速推理。

 

推理命令

 训练命令

 MIM命令管理

 模型配置文件

1. 通过修改模型配置文件可以直接对训练的模型结构进行改动。
2. 训练个人数据集主要改动num_classes为自己的分类数目。
3. 有了全局平均池化后，对图像的输入尺寸不再做固定要求。

 

 数据加载模块

1. 主要改动个人数据集路径。
2. 根据自己的GPU调节size大小。
3. 注意cpu的配比要适合batch size，避免出现瓶颈。

 

 数据预处理

根据不同训练任务调节训练时数据增强的方式，有利于提高模型的泛化能力。

 训练策略配置

        注意根据自己的GPU调节学习率，如官方8卡训练，lr0.01，自身单卡，我们本地训练lr就要设置为原来的1/8。