薛宇,曲永,闫好霖,王帅.基于注意力机制改进的YOLOv5绝缘子缺陷检测[J].电工技术,2024(17):185-189 |
基于注意力机制改进的YOLOv5绝缘子缺陷检测 |
|
|
DOI:10.19768/j.cnki.dgjs.2024.17.050 |
中文关键词: 绝缘子缺陷检测 注意力机制 CBAM YOLOv5s BiFPN |
英文关键词: |
基金项目: |
|
摘要点击次数: 77 |
全文下载次数: 0 |
中文摘要: |
提出了一种基于注意力机制改进的YOLOv5s检测方法,以提高绝缘子缺陷检测的准确率和效率。首先,在YOLOv5s网络模型的架构中,在其颈部(Neck)引入了空间与通道卷积块注意力模型(CBAM),目的是实现对缺陷区域的重点关注。同时,通过引入双向特征金字塔网络(BiFPN)进行优化改进,进一步完善了模型的性能。此外,对损失函数进行了优化,以进一步提高模型的整体训练效果。实验结果表明,改进的YOLOv5s算法相比原YOLOv5s算法对绝缘子缺陷检测的精准率提升了4.69%,召回率提升了2.56%,F1分数提升了4.09%,AP提升了5.16%。 |
英文摘要: |
|
查看全文 查看/发表评论 下载PDF阅读器 |