关键词:
无人机
输电线路巡检
遥感
目标检测
YOLOv5
摘要:
无人机已经成为高压输电线路检查的重要工具,而无人机拍摄的巡检影像通常包含复杂的背景和多种类型的小目标,这给目标检测算法带来诸多挑战。
为克服这些挑战,本文构建了一个基于真实无人机巡检影像的高质量输电线路巡检数据集,命名为TLID(Transmission Line Inspection Dataset),并在此基础上,设计了一种轻量级目标检测框架TLI-YOLOv5,为输电线路巡检任务提供了一种可靠的解决方案。本文的主要研究内容及创新点概述如下:
首先,构建高质量无人机输电线路巡检数据集TLID。TLID的原始影像从13744张真实的输电线路巡检影像中,遵循场景多样化和高清晰度的原则,挑选其中1231张影像进行标注,标注样本类别多达8种,包括杆塔、绝缘子、接线板、均压环、线夹、防振锤、塔牌和鸟巢,标注框数量达39,151个。这些影像覆盖我国多个省市的不同环境,如城市景观、农村环境、山地和森林地区。此外,数据集中的影像涵盖了多种不同距离、角度和天气条件,具有极高的多样性和复杂度,旨在提高模型的泛化能力。
其次,引入YOLOv5n至输电线路巡检领域,实现实时准确的输电线路目标检测。YOLOv5n作为YOLOv5系列中最小、最快的变体,其轻量化的程度甚至超过YOLOv6n和YOLOv8n,在精度和速度之间取得更好的平衡,因此更适合部署在实时输电线路巡检任务中。
最后,提出一种名为TLI-YOLOv5的轻量级目标检测框架,旨在实现更高效的输电线路目标检测。具体而言,在YOLOv5n网络的基础上,进行以下方面的改进:首先,将无参数注意力模块Sim AM融入YOLOv5n网络的骨干中,从而在不增加网络参数的情况下增强特征提取能力;其次,改进损失函数为Wise-Io U(WIo U),提高模型的准确度、鲁棒性和泛化能力;此外,还采用迁移学习和余弦学习率衰减的训练策略来加快训练过程中的收敛速度并提高模型的性能。
实验结果表明,与原始YOLOv5n相比,TLI-YOLOv5精度提高0.40%、召回率提高4.01%、F1分数提高1.69%、m AP50提高2.91%,以及m AP50-95提高0.74%,同时保持76.1FPS的高检测速度和4.15MB的紧凑体积。
本研究为大规模实时输电线路检查任务提供了一种实用高效的解决方案,旨在促进目标检测算法在输电线路巡检任务中的更广泛应用,推动输电线路巡检任务进一步走向自动化和智能化。