使用OpenMMLab开源目标检测算法库MMDetection,分别训练Faster R CNN和RTMDet的三角板目标检测算法。
在测试集上评估精度,预测图像、视频、摄像头实时画面。
目标检测预测框,用于后续Top-Down关键点检测算法。
以三角板关键点检测场景为例,结合OpenMMLab开源目标检测算法库MMDetection、开源关键点检测算法库MMPose、开源模型部署算法库MMDeploy,全面讲解项目全流程:
数据集:Labelme标注数据集、整理标注格式至MS COCO
目标检测:分别训练Faster R CNN和RTMDet-Tiny目标检测模型、训练日志可视化、测试集评估、对图像、摄像头画面预测
关键点检测:训练RTMPose-S关键点检测模型、训练日志可视化、测试集上评估、分别对“图像、视频、摄像头画面”预测
模型终端部署:转ONNX格式,终端推理