当前位置: 首页 > news >正文

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

http://www.lryc.cn/news/91815.html

相关文章:

  • Selenium自动化程序被检测为爬虫,怎么屏蔽和绕过
  • Nvidia Jetson Orin:开发技巧
  • 为什么需要 git 和 相关的小知识
  • (详解)vue中实现主题切换的三种方式
  • 英国皇家植物园采用机器学习预测植物抗疟性,将准确率从 0.46 提升至 0.67
  • 基于Locust实现MQTT协议服务的压测脚本
  • AURIX TC3XX Cached PFLASH与Non-Cached PFLASH的区别
  • uniapp开发小程序-显示左滑删除效果
  • FPGA 的数字信号处理:Verilog 实现简单的 FIR 滤波器
  • 使用粒子群优化算法(PSO)辨识锂电池二阶RC模型参数(附MATLAB代码)
  • 如何利用地面控制点实现倾斜摄影三维模型数据的几何坐标变换和纠正?
  • 设计规则之里氏替换原则
  • 【叠高高】叠蛋糕游戏的微信小程序开发流程详解
  • 收集关键词的方法有哪些?(如何查找精准的行业流量关键词)
  • 【GreenDao】RxQuery查询并修改GreenDao数据库,完成后更新UI
  • Modifier ‘public‘ is redundant for interface methods错误
  • Redis缓存击穿及解决问题
  • 环境感知算法——2.CenterNet基于KITTI数据集训练
  • JUC 高并发编程基础篇
  • 【十二】设计模式~~~行为型模式~~~命令模式(Java)
  • 可再生能源的不确定性和储能系统的时间耦合的鲁棒性和非预期性区域微电网的运行可行性研究(Matlab代码实现)
  • Revit中如何使创建的族文件内存变小
  • ClassLoader源码
  • Kafka分区消息积压排查指南
  • 数据库 期末复习(4) 概念数据库的设计
  • WuThreat身份安全云-TVD每日漏洞情报-2023-05-26
  • 关于Idea的一些常用设置项
  • Python使用WMI模块获取Windows系统的硬件信息,并使用pyinstaller库编译打包成exe的可执行文件
  • JavaScript语句(七)
  • 孪生诱捕网络在欺骗防御领域的应用