基于图神经网络的工井管孔智能识别方法及系统

本发明公开了基于图神经网络的工井管孔智能识别方法及系统，涉及计算机视觉与人工智能技术领域，包括以下步骤：采集相邻工井的图像并进行坐标系配准，提取管孔综合特征，以管孔为节点构建双工井异质图结构；将异质图结构输入多约束图神经网络，通过层次化消息传递机制聚合邻域信息，并结合空间连续性与布局一致性的综合损失函数约束优化，输出节点嵌入向量；基于节点嵌入向量计算管孔间相似度矩阵，融入领域知识约束，构建匹配代价矩阵；对匹配代价矩阵进行迭代优化，输出最优管孔匹配结果。本发明用以解决工井管孔依赖人工经验识别与匹配导致的效率低下、易出错且存在作业安全风险的问题。