MMPretrain代码课

安装注意事项

训练时需要基于算法库源码进行开发,所以需要git clone mmpretrain仓库。如果只调用,则pip install 即可。
在这里插入图片描述

from mmpretrain import get_model, list_models,inference_model

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

MMPretrain代码课_第1张图片
此时还没加载预训练权重

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

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

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

例如:图像分类任务,找到resnet18相关配置文件
MMPretrain代码课_第2张图片

便于后续修改
MMPretrain代码课_第3张图片
使用mmengine加载配置文件
MMPretrain代码课_第4张图片
修改类别数为2之后
MMPretrain代码课_第5张图片
数据集配置
MMPretrain代码课_第6张图片
MMpretrian中读取图像默认是用opencv做的,图像为bgr格式,需要转换。

此配置验证和测试不区分
MMPretrain代码课_第7张图片
schedule和优化器相关的操作
MMPretrain代码课_第8张图片
运行参数配置,日志打印间隔 interval
MMPretrain代码课_第9张图片
MMPretrain代码课_第10张图片

你可能感兴趣的:(计算机视觉,opencv,人工智能)