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
在这里插入图片描述
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值