基于注意力机制与特征平衡性的变电站多目标检测方法

摘要

基于注意力机制与特征平衡性的变电站多目标检测方法，包括：获取变电站设备的图像数据样本，利用数据标注软件提取图像数据样本中的目标信息，利用目标信息构建变电站多目标检测的数据集；基于darknet‑53网络和注意力机制，建立变电站多目标检测网络，对各特征图依次进行特征平衡和特征融合后得到融合特征图，再对融合特征图进行多分支检测，得到变电站多目标检测结果；使用变电站多目标检测的数据集，基于迁移学习的方法，对变电站多目标检测网络进行迭代训练；以训练好的多目标检测网络，对变电站设备进行检测。本发明解决具有复杂背景的目标与外形相似目标，以及不同目标相互遮挡导致识别难度大的问题，从而提高变电站运维的效率。