[发明专利]一种图像分割网络生成方法及图像分割方法

权利要求书

1.一种用于对双目图像中的人像进行分割处理的图像分割网络生成方法，适于在计算设备中执行，所述方法包括步骤：

构建分割处理网络，所述分割处理网络包括多个处理块、下采样层和上采样层；

构建视差计算网络，所述视差计算网络包括多个第一卷积层、转置卷积层和第二卷积层；

根据预先生成的第一图像数据集合对所述分割处理网络进行训练，根据预先生成的第二图像数据集合对所述视差计算网络进行训练，其中，所述第一图像数据集合包括多个第一图像组，每个第一图像组包括第一双目视图和对应的人像掩膜图，所述第二图像数据集合包括多个第二图像组，每个第二图像组包括第二双目视图和对应的视差图；

将训练好的分割处理网络和视差计算网络进行融合，以生成图像分割网络；

根据所述第一图像数据集合对所述图像分割网络进行训练，以便所述图像分割网络的输出指示输入图像所对应的人像掩膜图像，其中，训练后的所述图像分割网络可作为图像分割模型移植到移动终端。

2.如权利要求1所述的方法，所述构建分割处理网络的步骤包括：

构建处理块，所述处理块包括卷积层；

分别构建上采样层和下采样层；

按照预设的第一连接规则，将各处理块、下采样层和上采样层进行相连，以生成以一个处理块为输入、以另一个处理块为输出的分割处理网络。

3.如权利要求2所述的方法，所述构建处理块的步骤还包括：

分别构建激活层和图像叠加层；

按照预设的处理块构建规则，将一个或多个卷积层、激活层和图像叠加层进行相连，以生成以一个激活层为输入、以另一个激活层为输出的处理块。

4.如权利要求1-3中任一项所述的方法，所述视差计算网络还包括第一激活层和损失计算层，所述构建视差计算网络的步骤包括：

分别构建第一卷积层、转置卷积层和第二卷积层；

分别构建第一激活层和损失计算层；

按照预设的第二连接规则，将各第一卷积层、转置卷积层、第二卷积层、第一激活层和损失计算层进行相连，以生成以第一卷积层为输入、以第一激活层为输出的视差计算网络。

5.如权利要求1所述的方法，所述根据预先生成的第一图像数据集合对所述分割处理网络进行训练的步骤包括：

对每一个提取出的第一图像组，以该第一图像组所包括的第一双目视图为所述分割处理网络中第一个处理块的输入，以该第一图像组所包括的人像掩膜图为所述分割处理网络中最后一个处理块的输出，对所述分割处理网络进行训练。

6.如权利要求1所述的方法，所述根据预先生成的第二图像数据集合对所述视差计算网络进行训练的步骤包括：

对每一个提取出的第二图像组，以该第二图像组所包括的第二双目视图为所述视差计算网络中第一个第一卷积层的输入，以该第二图像组所包括的视差图为所述视差计算网络中最后一个第一激活层的输出，对所述视差计算网络进行训练。

7.如权利要求1所述的方法，所述将训练好的分割处理网络和视差计算网络进行融合，以生成图像分割网络的步骤包括：

对训练好的分割处理网络中的每一个下采样层，获取训练好的视差计算网络中、输出的特征图尺寸与该下采样层输出的特征图尺寸相等的第一卷积层，将该第一卷积层与该下采样层的输出进行合并，将合并结果作为与该下采样层相连的下一个处理块的输入；

对训练好的分割处理网络中的每一个上采样层，获取训练好的视差计算网络中、输出的特征图尺寸与该上采样层输出的特征图尺寸相等的转置卷积层，将该转置卷积层与该上采样层的输出进行合并，将合并结果作为与该上采样层相连的下一个处理块的输入，以生成图像分割网络。