关键词:
旋转目标检测
电表铅封螺钉
YOLOv8obb
多尺度特征融合
注意力机制
摘要:
针对单相智能电表自动化安装场景中,难以精确快速地定位防护盖铅封螺钉的问题,提出了一种基于改进YOLOv8obb的旋转检测算法GROOVE-YOLO,实现在单次检测中获取到目标螺钉的中心位置和旋转角度。首先,增加P2检测头,删除P5特征层,构建微小目标检测网络,提高模型对细小目标检测的能力,简化网络结构;其次,融合双向特征金字塔网络,引入学习权重,优化多尺度特征融合的效果,丰富模型的特征表达;最后,引入全局注意力机制,突出特征中关键信息,进一步提升检测精度。实验结果表明,改进算法的准确率、召回率和平均精度分别达到93.2%、90.5%和95.7%,与原模型相比分别提高了13.5%、17.1%和14.3%,平均中心点像素距离误差为1.4像素,平均旋转角度误差为4.8°,检测速度达到119 f/s。所提方法有效提高了防护盖铅封螺钉的定位精度,满足实际安装需求。