基于注意力模型和Soft-NMS的输电线路小目标检测方法

在输电线路的缺陷检测中,鸟巢以及塑料、碎布等挂空悬浮物多为小目标。其所占像素少,容易被背景干扰,检测精度有待提高。设计了一种全新的两阶段目标检测算法,用于改善对输电线路中鸟巢以及挂空悬浮物的检测效果。为了提高小目标检测的性能,在特征提取模块中融入注意力机制,以学习更为丰富的上下文信息。此外,在检测模块中,设计了基于更为柔和非极大值抑制算法的后处理方法,以减少小目标的丢失。与常用的两阶段目标检测算法相比,该方法在两个类别的平均准确率上分别提高了约4.7%和5.9%,有着更高的实际应用价值。