课程介绍
原始视频链接:
安装MMDetection和MMPose:https://www.bilibili.com/video/BV1Pa4y1g7N7
MMDetection三角板目标检测:https://www.bilibili.com/video/BV1Lm4y1879K
MMPose、RTMPose三角板关键点检测:https://www.bilibili.com/video/BV12a4y1u7sd
个人感觉这是一个非常成功的视频教学,完整的介绍了如何从原始数据训练一个可以简单定制的模型出来。当然,如果是纯粹的小白,可能就是一头雾水了。子豪哥这里按一下,那里按一下,然后结果就出来了,实战过程还是有点太快了。但是课程就是这样,不能满足所有人的需求,以下就是我看完课程后的一些笔记,后期我也会补上自己的模仿视频和代码。
配置环境
其实对于小白来说,这一步最困难,但是如果懂一点,基本上就不用在这里废话了,一路回车即可。
框架思路
回顾昨天讲的MMPose,这里用的是two stage的思路(或者叫Top Down,感觉不容易理解这个名词),就是先检测目标,然后预测关键点。
为了便于大家理解,大佬甚至先用人体姿态做了一个示范,所有模型的权重都提前训练好了。看下边代码截图就可以发现,用了两个模型权重。
大佬显示了一些结果,也是小黑子啊。
数据准备
看这个的话,其实数据准备就是下载大佬们标注的数据集,按照coco格式,包括目标检测框和关键点。这里直接看json还是非常明白的。
当然数据准备还包括一些什么删除清理复制之类的,不再赘述。
目标检测模型训练
感觉整个实验都已经非常傻瓜了,下载好训练脚本(这可不是配置文件),跑一下就行了。这里需要注意数据路径,必须按照大佬说的来。
结果还不错。
关键点模型训练
和上一个模型一样,下载训练脚本,一路回车即可。
结果也非常不错