基于YOLO目标检测算法的输电线路检测

摘要

输电线路是电网的重要组成部分,确保输电线路安全是保障电力系统稳定运行的重要措施。文章提出一种基于YOLO(you only Look once)算法的输电线路检测模型,用以对无人机等可见光拍摄图像中的输电线路进行物体检测。输电线路的识别检测任务一直是目标检测任务的难题,针对输电线路细长物理结构的特点,对微型目标检测YOLO模型进行改进,设计了对细长物理结构的高长宽比预测锚(anchor),对特征提取网络Darknet进行精简,在保持模型性能和检测精度的基础上对其进行轻量化处理,设计一种可通过单GPU(graphics processing unit)训练的轻量级模型并通过实验与原始YOLO模型、Faster R-CNN以及SSD在测试集上进行性能对比。实验结果表明,轻量级YOLO模型在测试集上检测精度达到83.04%,说明预测锚结构和精简后的网络模型对于输电线路检测是有效的。