科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对输电线路巡检的复杂场景下被遮挡以及目标较小导致的误检问题，文中提出了一种基于YOLOv5的改进算法模型。首先通过数据增强和数据扩充数据集进行预处理；其次给引入SE模块，加强对不同尺度下目标的特征融合；然后引入CBAM模块，进行进一步特征提取，使提取的特征信息更加突出。最后，对损失函数进行优化，改进小数据导致的样本不均衡现象。经实验证明，改进后的算法有效地提高了被遮挡的目标和小目标的识别率，mAP较原算法精度提高了5%,Recall提高了14.6%。与其他改进模型相比，精度、召回率都有提高，验证了该模型在各种场合下具有较强的鲁棒性和泛化能力。
- 单位
