基于递归图神经网络解决边缘过平滑的方法及装置

摘要

本发明公开了一种基于递归图神经网络解决边缘过平滑的方法及装置，方法包括以下步骤：利用图神经网络建模边缘位置特征，采用更新后的超像素特征来增强边缘像素的表示，并将增强后的边缘像素映射回原规则网格中；利用图神经网络建模所述边缘位置特征和像素关系，利用超像素的特征来增强边缘像素的表示，重新计算得到该些位置的分类结果，递归图传播边缘优化直到输出的分割图的分辨率与原始图像相同。装置包括：处理器和存储器。本发明利用图神经网络建模边缘位置特征与区域超像素特征的关系、超像素与超像素特征之间的关系，更好地利用边缘像素的上下文信息实现更精确的边缘像素分类。