[发明专利]通过图神经网络的链路预测进行血管分割的方法和装置

权利要求书

1.一种通过图神经网络的链路预测进行血管分割的方法，其特征在于，包括：

将待测血管图像输入至U-net模型，获取由所述U-net模型输出的目标第一邻接矩阵；

基于所述待测血管图像的节点集、边集和所述目标第一邻接矩阵，生成目标血管连通图；

其中，所述U-net模型为，以带边标签的样本血管图像为训练样本，基于目标损失函数，训练得到。

2.根据权利要求1所述的通过图神经网络的链路预测进行血管分割的方法，其特征在于，在所述将待测血管图像输入至U-net模型之前，所述方法包括：

基于所述样本血管图像的节点集和边集，生成样本第一邻接矩阵；

基于节点特征和所述样本第一邻接矩阵，预测节点之间的链路，生成第二邻接矩阵；

基于所述样本第一邻接矩阵和所述第二邻接矩阵，确定所述目标损失函数。

3.根据权利要求2所述的通过图神经网络的链路预测进行血管分割的方法，其特征在于，所述基于所述样本血管图像的节点集和边集，生成样本第一邻接矩阵，包括：

对所述节点集中目标节点进行聚类，生成与所述目标节点对应的簇；

基于所述目标节点对应的簇和所述簇中每个像素点的特征，生成所述目标节点对应的节点特征；

基于所述节点特征之间的余弦相似度，生成所述样本第一邻接矩阵。

4.根据权利要求2所述的通过图神经网络的链路预测进行血管分割的方法，其特征在于，所述基于节点特征和所述样本第一邻接矩阵，预测节点之间的链路，生成第二邻接矩阵，包括：

基于图卷积对所述节点特征进行编码，生成节点嵌入表示；

使用内积对所述节点嵌入表示进行解码，生成所述第二邻接矩阵。

5.根据权利要求4所述的通过图神经网络的链路预测进行血管分割的方法，其特征在于，所述使用内积对所述节点嵌入表示进行解码，生成所述第二邻接矩阵，包括：

基于公式：

with Z＝Encoder(f,A)

生成所述第二邻接矩阵，其中，为所述第二邻接矩阵，Z为所述节点嵌入表示，f为所述节点特征，A为所述样本第一邻接矩阵，Encoder为目标图卷积层，σ为sigmoid函数。

6.根据权利要求2所述的通过图神经网络的链路预测进行血管分割的方法，其特征在于，在所述基于所述样本血管图像的节点集和边集，生成样本第一邻接矩阵之前，所述方法包括：

基于所述样本血管图像中血管的人工注释，生成所述血管的骨架结构；

在所述骨架结构上进行角点采样，生成所述节点集；所述节点集包括角点，各所述角点间欧几里德距离不低于第一目标阈值；

基于所述血管的人工注释，生成所述节点集中两个所述节点之间的测地距离；

在所述测地距离小于第二目标阈值的情况下，在所述测地距离对应的节点之间构造边，生成所述边集；

基于所述骨架和所述边集之间的距离，确定所述边集中目标边的正负值。

7.一种通过图神经网络的链路预测进行血管分割的装置，其特征在于，包括：

第一处理模块，用于将待测血管图像输入至U-net模型，获取由所述U-net模型输出的目标第一邻接矩阵；

第二处理模块，用于基于所述待测血管图像的节点集、边集和所述目标第一邻接矩阵，生成目标血管连通图；

其中，所述U-net模型为，以带边标签的样本血管图像为训练样本，基于目标损失函数，训练得到。

8.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至6任一项所述通过图神经网络的链路预测进行血管分割的方法。

9.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所通过图神经网络的链路预测进行血管分割的方法。

10.一种计算机程序产品，包括计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述通过图神经网络的链路预测进行血管分割的方法。