高级检索
苏晓, 张明晖, 陈峻宇, 丁争, 许华栋, 白万崧. 基于集成YOLOv5算法的输电线路杆塔目标检测[J]. 农村电气化, 2023, (5): 33-39. DOI: 10.13882/j.cnki.ncdqh.2023.05.009
引用本文: 苏晓, 张明晖, 陈峻宇, 丁争, 许华栋, 白万崧. 基于集成YOLOv5算法的输电线路杆塔目标检测[J]. 农村电气化, 2023, (5): 33-39. DOI: 10.13882/j.cnki.ncdqh.2023.05.009
SU Xiao, ZHANG Minghui, CHEN Junyu, DING Zheng, XU Huadong, BAI Wansong. Target Detection of Transmission Line Towers Based on Integrated YOLOv5 Algorithm[J]. RURAL ELECTRIFICATION, 2023, (5): 33-39. DOI: 10.13882/j.cnki.ncdqh.2023.05.009
Citation: SU Xiao, ZHANG Minghui, CHEN Junyu, DING Zheng, XU Huadong, BAI Wansong. Target Detection of Transmission Line Towers Based on Integrated YOLOv5 Algorithm[J]. RURAL ELECTRIFICATION, 2023, (5): 33-39. DOI: 10.13882/j.cnki.ncdqh.2023.05.009

基于集成YOLOv5算法的输电线路杆塔目标检测

Target Detection of Transmission Line Towers Based on Integrated YOLOv5 Algorithm

  • 摘要: 杆塔是输电线路中的重要组成设施,其安全直接影响电网电力输送的安全稳定。根据遥感影像中杆塔小目标识别精度低等问题,研究基于YOLOv5s和YOLOv5x算法进行集成建模,并加入加权框融合(weighted boxes fusion,WBF)推理机制,借助高分辨率遥感杆塔影像数据集进行模型训练测试,并对数据集做测试时增强(test-time augmentation,TTA)。实验结果显示:与单模型识别结果相比较,集成YOLOv5模型识别精确度、召回率、mAP@.5显著提升,分别达到0.952、0.944、0.929;并且在一些复杂背景、不同光照环境和不同天气条件下模型都具有良好的识别效果,具有较强的鲁棒性。

     

    Abstract: Transmission tower is an important component of transmission line, and its safety directly affects the safety and stability of power transmission. According to the low accuracy of small target identification of transmission tower in remote sensing images, in this study, integrated modeling was conducted based on YOLOv5s and YOLOv5x algorithms, weighted boxes fusion (WBF) reasoning mechanism was added, model training was conducted with high-resolution remote sensing tower image data set, and performed a test-time augmentation of the data set. The experimental results showed that compared with the single model recognition results, the recognition accuracy, recall rate and mAP@.5 of integrated YOLOv5 model are significantly improved, reaching 0.952, 0.944 and 0.929 respectively. In addition, under some complex background, different illumination environment and different weather conditions, the model proposed in this paper has good recognition effect and strong robustness.

     

/

返回文章
返回