前言:Hello大家好,我是小哥谈。现有的从无人机(UAV)上进行的绝缘子缺陷识别方法在处理复杂背景场景和小型物体时,准确率较低且存在较多的假阳性检测。为解决这一问题,本文提出了一种基于局部注意力建模的新注意力基础架构,即YOLO-ELA。在YOLOv8的一阶段架构的Neck部分添加了高效局部注意力(ELA)模块,将模型的注意力从背景特征转向具有缺陷的绝缘子特征。🌈
目录
🚀1.基础概念
🚀2.网络结构
🚀3.添加步骤
🚀4.改进方法
💥💥步骤1:新建ELA.py文件
💥💥步骤2:修改tasks.py文件
💥💥步骤3:创建自定义yaml文件
💥💥步骤4:新建train.py文件