基于图神经网络的变电站场景三维目标检测

来源：计算机工程与应用杂志2023年第9期北京时间：

作者：张婷, 张兴忠, 王慧民, 杨罡, 王大伟

单位：1.太原理工大学 软件学院，山西 晋中 030600；2.国网山西省电力公司 电力科学研究院，山西 晋中 030600

摘要：在变电站三维场景中，对巡检人员和带电设备的精确定位与识别是提高人员安全管控水平的前提。针对变电站复杂场景中目标定位与识别不准的问题，提出了一种基于图神经网络的变电站场景三维目标检测方法。该方法基于point-GNN结构设计，在顶点特征提取阶段，提出PCS（point-channel-sphere）注意力结构，提取更加丰富的关键点特征信息；在GNN边缘特征聚合阶段，采用统筹性池化机制，兼顾最大池化和均值池化从而获取更丰富的全局特征；改进模型损失函数，将Focal Loss作为分类损失使训练更加关注前景点，将DIoU Loss作为回归损失使回归任务更高效。在自建的变电站场景数据集上进行训练与测试，实验表明该方法mAP值达到73.81%，优于基准模型，能够改善变电站场景中目标的检测效果，对提高人员安全管控水平具有一定的实用价值。

关键词：图神经网络,三维目标检测,点云,注意力,统筹性池化,损失函数