简介
整个项目的模型采用属于Top-down范式的RTMPose模型,基于MMDetection和MMPose框架,采用三角板关键点检测数据集实现三角板的目标检测和关键点检测。
参考教程
安装MMDetection和MMPose
参考视频教程,在本地Win11(显卡:RTX 2070)完成环境配置:
MMDetection三角板目标检测
流程
- 数据集下载(采用MS COCO格式标注)
- 下载目标检测的config配置文件
- 模型训练
- 可视化训练日志
- 模型权重文件精简转换
- 下载训练好的模型权重
- 利用下载的预训练模型进行图像和视频的检测
解释:权重的精简转换可以让.pth权重文件大小缩小为原来的一半以上,但不影响推理结果和推理速度。
MMPose、RTMPose三角板关键点检测
流程
- 下载关键点检测的config配置文件
- 训练RTMPose
- 可视化训练日志
- 模型权重文件精简转换
- 下载训练好的模型权重
- 三角板关键点的预测(推理)
思考:关键点热力图可视化发现关键点是等间距排布的,这可能RTMPose模型的一个内在特性,需要读论文探究一下。