论文检索
期刊
全部知识仓储预印本开放期刊机构
高级检索

基于改进YOLOv7的输电铁塔塔基检测算法OA北大核心CSTPCD

A novel algorithm based on the improved YOLOv7 for detecting transmission tower base

中文摘要英文摘要

输电塔作为整个电力传输系统最重要的组成部分之一,需要及时对输电塔进行检测保证塔基的稳固以保障后期的使用.针对无人机采集到的输电塔图像存在背景复杂、背景与目标塔基对比度低、小目标及塔基不完整等问题,提出了基于改进YOLOv7 的输电塔塔基检测算法.首先,通过无人机采集不同地形地貌的输电塔图像,构建高质量数据集.然后,在原始YOLOv7 的Back-bone层中加入卷积注意力模块CBAM注意力机制,以提高输电塔塔基特征的提取能力.最后,引入 WIoU v3 代替原坐标损失函数CIoU,以提高目标检测任务的准确性和稳定性.在该数据集上,使用改进后的YOLOv7 算法与目前主流的目标检测算法进行对比实验,实验结果中所提算法的mAP值高达 99.93%,比原始YOLOv7 提高 2.19%,FPS值为 37.125,满足实时检测需求,算法的整体性能较好.实验验证了所提算法在塔基检测上的可行性和有效性,为后续塔基区周围水土情况的研究奠定了基础.

The pylon is one of the most important components in the entire power transmission system.It is necessary to timely inspect the tower to ensure the stability of the base for the later use.There are problems of the transmission tower images collected by UAV have complex backgrounds,the background is similar to the base of target tower,as well as small objects and incomplete tower base,this paper proposes an improved YOLOv7 algorithm for detecting the base of tower.Firstly,using the pylon images of different landforms to construct high-quality data sets.Then CBAM attention mechanism is added to the Backbone layer of the original YOLOv7 to improve the feature extraction ability of the pylon.Finally,introducing WIoU v3 instead of the original coordinate loss function CIoU to improve the veracity and stability of target detection tasks.On this dataset,a comparative experiment was conducted using the improved YOLOv7 algorithm and the current mainstream object detection algorithm.The mAP value of our algorithm is as high as 99.93%in the experimental results,it is 2.19%higher than the original YOLOv7,the FPS value is 37.125,which meets the real-time detection requirements,and the overall performance of the algorithm is good.It's feasible and effective in detection tasks of towers'base for our algorithm,which has been proven by the experiments in this paper,and laying the foundation for future research on the soil and water around the base of tower.

雷磊;魏小龙;梁俊;董倩;肖樟树

国网陕西省电力有限公司 电力科学研究院,陕西 西安 710100||国网(西安)环保技术中心有限公司,陕西 西安 710100国网陕西省电力有限公司,陕西 西安 710048陕西师范大学 计算机科学学院,陕西 西安 710119

计算机与自动化

输电塔塔基;YOLOv7;目标检测;卷积块注意力模块;WIoU v3

transmission tower base;YOLOv7;object detection;convolutional block attention module(CBAM);WIoU v3

《陕西师范大学学报(自然科学版)》 2024 (003)

85-95 / 11

陕北地区电网工程水土流失及次生灾害风险识别与治理关键技术研究与应用(5226KY22000K);国家自然科学基金(61672333)

10.15983/j.cnki.jsnu.2024012

评论

下载量:0
点击量:0