[发明专利]一种图像分割方法及电子设备

说明书

本申请公开一种图像分割方法及电子设备，包括：将待分割图像输入预设的分割网络；利用所述分割网络对所述待分割图像进行分割处理，以从所述待分割图像中确定目标物图像；所述分割网络通过预设的辅助网络和几何模块训练得到；所述几何模块用于确定非标注样本对应的几何特征图；利用分割网络实现图像分割处理，分割网络通过预设的辅助网络和几何模块进行训练，几何模块用于确定非标注样本对应的几何特征图；所以分割网络能够针对目标物共性的形状和结构进行特征提取，从而对该形状和结构加以有效的利用，提高了分割网络进行图像分割处理的效率和准确性。

技术领域

本申请涉及计算机视觉技术领域，尤其涉及一种图像分割方法及电子设备。

背景技术

当前，神经网络技术已经在图像识别领域得到广泛的运用。而图像分割，正是图像识别领域中一项重要的具体应用。图像分割即利用图像分割网络将图像中特定的目标物，与目标物以外的背景部分分割开来，以便于后续针对目标物部分进行进一步处理。

在一些领域当中，图像中特定的目标物具有较为明确的几何形状和结构特点。不过现有的分割网络，并未对此形状和结构加以有效的利用，不能够针对目标物共性的几何形状和结构特点进行针对性的特征提取，从而实现图像分割，可见此类分割网络的性能和效率，尚有待提高。

发明内容

本申请提供一种图像分割方法，包括：

将待分割图像输入预设的分割网络；

利用所述分割网络对所述待分割图像进行分割处理，以从所述待分割图像中确定目标物图像；

所述分割网络通过预设的辅助网络和几何模块训练得到；所述几何模块用于确定非标注样本对应的几何特征图。

优选的，所述待分割图像包括，医学图像；所述利用所述分割网络对所述待分割图像进行分割处理，以从所述待分割图像中确定目标物图像包括：

利用所述分割网络的编码网络，确定所述医学图像对应的待分割特征图；

利用所述分割网络的解码网络，确定所述待分割特征图的图像特征；

利用所述分割网络的结构识别网络，确定所述待分割特征图对应的结构特征；

根据所述图像特征和所述结构特征，确定所述目标物图像；所述目标物图像中，包括所述医学图像中的器官结构图像。

优选的，所述结构识别网络通过所述几何模块确定的几何特征图训练得到；所述几何特征图对应特定的器官结构。

优选的，所述通过预设的辅助网络和几何模块训练得到所述分割网络包括：

确定标注样本的标注标签；利用待训练网络确定所述标注样本的第一预测标签；

利用所述待训练网络、所述辅助网络和所述几何模块，确定所述非标注样本对应的几何特征图；

根据所述标注样本的标注标签、所述标注样本的第一预测标签和所述非标注样本对应的几何特征图，对所述待训练网络进行数据训练；

将完成所述数据训练的所述待训练网络确定为所述分割网络。

优选的，所述利用所述待训练网络、所述辅助网络和所述几何模块，确定所述非标注样本对应的几何特征图包括：

利用所述待训练网络和所述几何模块，确定所述非标注样本对应的第一几何特征图；

确定所述非标注样本对应的扰动样本；

利用所述辅助网络和所述几何模块，确定所述扰动样本对应的第二几何特征图。

优选的，所述利用所述待训练网络和所述几何模块，确定所述非标注样本对应的第一几何特征图包括：