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

摘要

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

关键词

注意力机制 / Soft-NMS / 小目标检测 / 输电线路