OpenMMLab AI笔记9——MMPose代码实战课

本文记录了一次OpenMMLab AI实战课程,内容涉及MMDetection和MMPose的使用,以耳朵穴位检测为例,包括环境配置、目标检测和关键点检测的训练与预测,强调了理解和知识迁移的重要性。
摘要由CSDN通过智能技术生成

以下文字仅是一个入门小白的课堂笔记分享,如有不妥之处,恳请各位朋友指出!


本次实战课教程以三角板关键点检测数据集为例,分别从相关算法库的安装、MMDetection三角板目标检测和MMPose关键点检测三部分着手讲解,对应的MMPose作业是基于RTMPose的耳朵穴位关键点检测,任务包括在已标注好的耳朵穴位数据集上使用MMDetection算法库训练RTMDet耳朵目标检测算法,提交测试集评估指标;使用MMPose算法库,训练RTMPose耳朵关键点检测算法,提交测试集评估指标;用自己的耳朵进行图像预测和视频预测,并提交测试集评估指标。

在github上主讲人子豪兄准备了细致的教程,但由于数据集不同,盲目的模仿操作并不能是我们直接完成作业,更多地要求我们尽量理解每一步骤代码的作用原理,将过程内化为自己的思路,再灵活地运用在其它数据集或自己标注的数据集上。这种知识迁移模式是这次实战营的一个亮点,于我也是一个挑战。

下面我用自己的话来概况一下过程吧。因为本次采用自上向下方法,关键点预测是在目标检测任务上实现的,就是说,对一张图片第一步先要目标检测,将主体用锚框框出来,第二步再在锚框上进行关键点预测,所以很多同样的步骤要分别在MMDetection和MMPose上操作两次。

首先是配置环境,我的云GPU环境是GPU RTX 3060、CUDA v11.2,安装Pytorch==1.10.0,通过pip安装openmim,再通过mim命令分别安装mmengine、mmcv和mmdet,但这里要特别注意,安装mmdet时经常会报错版本问题(我在AutoDL上操作的,不知道其它平台会不会这样),除此之外还需安

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值