标题:基于改进YOLOv8的输电线路异物检测算法研究 作者:宋义 丁革媛 作者单位:沈阳工业大学化工过程自动化学院,辽宁 辽阳 111000 关键字:输电线路;异物检测;YOLOv8;CBAM;可变形卷积;EIOU 摘要:针对目前输电线路的异物检测存在效率低、难度大、检测成本高等问题,提出了一种基于改进YOLOv8算法的输电线路异物检测的方法。首先,在原有的基础结构上添加注意力机制CBAM模块,利用注意力机制来增强网络的表示能力,可以将注意力集中在更为重要的关键特征上,提高对小目标的注意力;其次,针对异物形状不同具有多样性的问题,引入可变形卷积,使网络可以增强对发生形变的物体的学习能力,且可以改善小目标物体检测存在感受野不足的问题;最后,在模型输出层检测头部分使用EIOU损失函数作为边框回归损失函数。改进后的算法在自建的输电线路异物检测数据集中取得了良好的效果,与原模型相比,mAP提升了2.39%,精确率、召回率均有所提升。 |