安装注意事项
训练时需要基于算法库源码进行开发,所以需要git clone mmpretrain仓库。如果只调用,则pip install 即可。
from mmpretrain import get_model, list_models,inference_model
分别用于模型的获取、例举、推理
此时还没加载预训练权重
torchvision中也有数据集和预训练模型,为什么还要选mmpretrian?
- 支持分布式训练
- 训练中断后恢复训练
- 训练脚本中有很多功能
- 保存每次训练的参数
图像分类任务配置文件详解
例如:图像分类任务,找到resnet18相关配置文件
便于后续修改
使用mmengine加载配置文件
修改类别数为2之后
数据集配置
MMpretrian中读取图像默认是用opencv做的,图像为bgr格式,需要转换。
此配置验证和测试不区分
schedule和优化器相关的操作
运行参数配置,日志打印间隔 interval