面向输电线路小目标异物检测的改进YOLO网络

作者:邹辉军; 焦良葆*; 张智坚; 汤博宇; 刘子恒
来源:南京工程学院学报(自然科学版), 2022, 20(03): 7-14.
DOI:10.13960/j.issn.1672-2558.2022.03.002

摘要

为解决输电线路所在环境较复杂多变、线路上异物目标体积较小难以识别等问题,提出将BCA-YOLO网络针对小目标检测进行优化,将YOLO v5中的CSP2_X替换为CSP_CA,再添加一层小目标检测层,将原网络中的FPN结构替换为计算量小的BiFPN;针对一般输电线路异物数据集中图片较少的问题,提出通过场景增强、Mixup和加入噪声模拟对数据集进行有效扩充.试验结果证明,相较于传统的YOLO v5网络,mAP_0.5提高了3.8%、查全率提升6%、查准率提高了6.1%,更好地满足了隐患检测的工程实际需求.

全文