第三讲 MMPose代码教程

该项目基于MMDetection和MMPose框架,利用RTMPose模型进行三角板目标检测和关键点检测。通过三角板关键点检测数据集,完成模型训练并进行权重精简转换。文章还提到了关键点等间距排布的现象,暗示了RTMPose模型的潜在特性。
摘要由CSDN通过智能技术生成

简介

整个项目的模型采用属于Top-down范式的RTMPose模型,基于MMDetection和MMPose框架,采用三角板关键点检测数据集实现三角板的目标检测和关键点检测。

参考教程

保姆级代码教程

安装MMDetection和MMPose

参考视频教程,在本地Win11(显卡:RTX 2070)完成环境配置:
环境依赖Anaconda3,配置为mmpose虚拟环境

MMDetection三角板目标检测

流程

  1. 数据集下载(采用MS COCO格式标注)
  2. 下载目标检测的config配置文件
  3. 模型训练
  4. 可视化训练日志
  5. 模型权重文件精简转换
  6. 下载训练好的模型权重
  7. 利用下载的预训练模型进行图像和视频的检测
    解释:权重的精简转换可以让.pth权重文件大小缩小为原来的一半以上,但不影响推理结果和推理速度。

MMPose、RTMPose三角板关键点检测

流程

  1. 下载关键点检测的config配置文件
  2. 训练RTMPose
  3. 可视化训练日志
  4. 模型权重文件精简转换
  5. 下载训练好的模型权重
  6. 三角板关键点的预测(推理)
    思考:关键点热力图可视化发现关键点是等间距排布的,这可能RTMPose模型的一个内在特性,需要读论文探究一下。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值