摘要
传统的电力工人作业安全关键装备检测算法精度低、鲁棒性差.为此提出了一种基于YOLOv5的改进型目标检测算法,用于对绝缘手套、安全帽、作业人员进行检测.首先在原版YOLOv5中加入坐标注意力模块,提升对小目标特征提取的能力;其次用加权双向特征金字塔网络结构代替原有的特征金字塔网络结构,进一步提升特征提取的能力.实验结果表明平均精确度提升了1.8%,其精度达96.4%,平均精确度均值提升了0.4%,其均值达93.3%.所提算法改善了原版网络对小目标容易漏检、误检的问题,具有较强的实用性和先进性,能满足实时检测的要求,对电力行业安全有一定的促进作用.
- 单位