[发明专利]一种图像分割方法及装置

说明书

本发明提供一种图像分割方法及装置，利用预设的3D卷积神经网络模型确定获取的待分割图像进行参数调整后的每个目标物的分割图像，利用3D卷积神经网络模型中的提取模块提取待分割图像的至少一个目标物的第一特征图矩阵，像素级显著性增强模块对每个目标物的第一特征图矩阵的参数进行调整，确定每个目标物的像素级的加权矩阵，根据通道级显著性增强模块增强每个目标物的第一特征图矩阵的矩阵通道，确定每个目标物的通道级的加权矩阵，3D残差反卷积模块对增大每个目标物的像素级的加权矩阵和通道级的加权矩阵的和得到目标物的目标矩阵的尺寸进行还原处理，确定进行参数调整后的每个目标物的分割图像。基于本发明，能够得到高精度的分割图像。

技术领域

本发明涉及计算机视觉技术领域，尤其涉及一种图像分割方法及装置。

背景技术

随着社会的快速发展，计算机视觉技术的发展也越来越快，尤其是计算机视觉技术中的图像分割技术。其中，图像分割是指将图像分成若干具有相似性质的区域的过程。

现有的图像分割技术是通过将图像输入传统的2D卷积神经网络中，通过卷积操作提取图像的特征图，再对提取到的特征图进行还原得到分割结果，即分割图像。但是，基于传统的2D卷积神经网络对图像进行分割得到的分割图像的精度低。

发明内容

有鉴于此，本发明的目的在于提供一种图像分割方法及装置，以提高分割图像的精度。

为了实现上述目的，本发明提供如下技术方案：

本发明实施例第一方面公开了一种图像分割方法，该方法包括：

获取待分割图像；

利用预设的3D卷积神经网络模型确定所述待分割图像中进行参数调整后的每个目标物的分割图像，所述3D卷积神经网络模型基于图像样本数据预先训练得到，所述3D卷积神经网络模型包括：提取模块、像素级显著性增强模块、通道级显著性增强模块和3D残差反卷积模块；具体的利用所述预设的3D卷积神经网络模型确定所述待分割图像中进行参数调整后的每个目标物的分割图像的过程包括：

利用所述提取模块，提取所述待分割图像的至少一个目标物的第一特征图矩阵；

利用所述像素级显著性增强模块，对每个所述目标物的第一特征图矩阵中的参数进行调整，确定每个所述目标物的像素级的加权矩阵，其中，所述每个所述目标物的第一特征图矩阵中的参数为所述目标物的像素；

根据所述通道级显著性增强模块，增强每个所述目标物的第一特征图矩阵中的矩阵通道，确定每个所述目标物的通道级的加权矩阵；

利用所述3D残差反卷积模块，针对于每个目标物，计算该目标物的像素级的加权矩阵和通道级的加权矩阵的和，得到所述目标物的目标矩阵，增大每个所述目标物的目标矩阵的尺寸，并对每个目标物的进行尺寸增大后的目标矩阵还原处理，确定所述待分割图像中进行参数调整后的每个所述目标物的分割图像。

可选的，所述利用所述像素级显著性增强模块，对每个所述目标物的第一特征图矩阵中的参数进行调整，确定每个所述目标物的像素级的加权矩阵，包括：

针对于每一个目标物，利用所述像素级显著性增强模块对所述目标物的特征图矩阵进行维度变换、维度调整以及非线性处理，得到所述目标物的第二特征图矩阵；

将每一个目标物的第一特征图矩阵与所述目标物的第二特征图矩阵进行加权相加，得到每个所述目标物的像素级的加权矩阵。

可选的，所述根据所述通道级显著性增强模块，增强每个所述目标物的第一特征图矩阵中的矩阵通道，确定每个所述目标物的通道级的加权矩阵，包括：

针对与每一个目标物，利用所述通道级显著性增强模块对所述目标物的第一特征图矩阵进行维度变换、维度调整以及非线性处理，得到每个目标物的第三特征图矩阵；