标题:基于YOLOv5s的绝缘子识别算法研究 作者:俞炜平1 李振海1;高亚洲2 陈艳芳2 杨鹤猛2;刘志鹏1 作者单位:1 国网福建省电力有限公司莆田供电公司,福建 莆田 351100;2 天津航天中为数据系统科技有限公司,天津市智能遥感信息处理技术企业重点实验室,天津 300301 关键字:绝缘子;YOLOv5s;注意力机制;轻量化网络;空间金字塔池化 摘要:为提高绝缘子的检测精度,提出一种改进的YOLOv5s绝缘子检测算法。结合多尺度思想,设计一种新的注意力模块MECA,避免浅层信息丢失,提高识别精度;采用GSConv和VoV-GSCSP替换网络中的普通卷积和C3模块,使模型更加轻量化;利用SimSPPF替换SPPF,并将SimSPPF的ReLU函数替换成Mish函数,解决图片失真问题;最后对模型损失函数进行优化,解决了边框回归不准确问题。实验表明,该网络的平均精度均值(mAP)提高了2.1%,检测精度提高了0.8%,参数计算量减少了13.3%,取得不错的效果。 |