今天讲解的主要是实际的实操过程
part1 MMClassification
MMClassification是图像分类的工具包,支持大量的图像分类模型
MMClassfication仓库地址:https://github.com/open-mmlab/mmclassification
MMClassfication教程地址:https://mmclassification.readthedocs.io/en/latest/
part2 配置文件
深度模型训练涉及以下几个方面
模型结构:模型层数、每层通道数等
数据集:数据集类型、数据文件划分、文件路径、数据增强策略等
训练策略:梯度下降算法、学习略参数、batch_size、训练总伦次、学习略变化策略
运行时:GPU、分布式环境配置
辅助功能:打印日志、定时保存checkpoint等等
OpenMMLab项目中,所有项目都涵盖在一个配置文件中,一个配置文件定义了一个完整的训练过程
model字段定义模型
data字段定义数据
optimizer、lr_config字段定义训练策略
load_from字段定义与训练模型的参数文件