MMPretrain代码课

安装注意事项

训练时需要基于算法库源码进行开发，所以需要git clone mmpretrain仓库。如果只调用，则pip install 即可。

from mmpretrain import get_model, list_models,inference_model

分别用于模型的获取、例举、推理

此时还没加载预训练权重

torchvision中也有数据集和预训练模型，为什么还要选mmpretrian？

• 支持分布式训练
• 训练中断后恢复训练
• 训练脚本中有很多功能
• 保存每次训练的参数

图像分类任务配置文件详解

例如：图像分类任务，找到resnet18相关配置文件

便于后续修改

使用mmengine加载配置文件

修改类别数为2之后

数据集配置

MMpretrian中读取图像默认是用opencv做的，图像为bgr格式，需要转换。

此配置验证和测试不区分

schedule和优化器相关的操作

运行参数配置，日志打印间隔 interval