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

作     者：张婷 张兴忠 王慧民 杨罡 王大伟 ZHANG Ting;ZHANG Xingzhong;WANG Huimin;YANG Gang;WANG Dawei

作者机构：太原理工大学软件学院山西晋中030600 国网山西省电力公司电力科学研究院山西晋中030600 

基　　金：山西省重点研发计划项目(201803D31041) 国网山西省电力公司信息化项目(B1051C200016) 

出 版 物：《计算机工程与应用》 (Computer Engineering and Applications)

年 卷 期：2023年第59卷第9期

页      码：329-336页

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

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

学科分类：081203[081203] 08[工学] 0835[0835] 0812[工学-测绘类] 

核心收录：

D　O　I：10.3778/j.issn.1002-8331.2111-0264

馆 藏 号：203121369...