[发明专利]一种基于卷积神经网络的同类前景图像集分割方法

权利要求书

1.一种基于卷积神经网络的同类前景图像集分割方法，其特征在于，包括：通过计算机对图像做如下步骤的处理；

步骤1，向计算机输入同类图像集同类图像集；由计算机从同类图像集中读取一张，生成样本分割掩码；

步骤2，构建训练数据集；

步骤3，学习分割网络模型；

步骤4，基于分割网络模型分割同类图像集中的其余图像；

步骤5，进行是否需要再分割的判断；循环的条件是是否所有图像都已取得满意的分割结果；

如果所有图像均已获得满意的分割结果，则分割结束；

反之，选择分割效果较差的图像作为新的分割样本，并返回步骤1训练新的分割网络模型，实现对其余分割欠佳图像的分割优化。

2.如权利要求1所述一种基于卷积神经网络的同类前景图像集分割方法，其特征在于：具体步骤依次为：

步骤1，生成样本分割掩码：从给定的同类图像集中随机选取1幅图像，采用交互式分割方法进行手动分割，得到分割掩码图像；同类图像集图像集中含有n幅图像；

步骤2，构建训练数据集：对步骤1得到的分割掩码图像进行几何变换；所述几何变换包括仿射变换和薄板样条变换，所述仿射变换是指对图像进行缩放、平移和旋转，薄板样条变换是指利用薄板样条函数(Thin Plate Spline，TPS)对要变换的形状进行插值,它是一种非刚性变换；选取10²至10³种组合生成10²至10³张不同的变换掩码，模拟前景物体的10²至10³种姿态变换，将变换后的掩码与对应原始图像结合，构成10²至10³个4通道图像；

再将10²至10³个4通道图像与由步骤1得到的分割掩码图像配对，构成10²至10³对训练数据；其中，每个4通道图像均包含一张待分割图像和一张待分割前景变换后的掩码轮廓图，分割掩码图像给出了待分割前景的真实轮廓；

步骤3，学习分割网络模型：以步骤2中的所有4通道图像，大概有10²至10³个，作为网络的输入，经过10至30层的卷积层、池化层、激活层、分类输出层及上采样层的逐次处理后，获得“上采样层的特征图”；优选的，卷积层为20层；

在上采样层的特征图上进行逐像素分类，并与对应的分割掩码图像进行比较得到预测误差；

采用反向传播算法和随机梯度下降法来减小预测误差以训练该神经网络,得到同类前景图像集分割的网络模型；所述反向传播算法是指神经网络中最有效的方法，其主要思想是将网络最后输出的结果，和真实结果对比，计算其误差，并将误差反向逐级传下去；所述随机梯度下降法是指在更新网络参数的时候，并不遍历整个数据集，是一种批处理方法，每次更新参数后，都能够让损失函数变小，最终达到最小或者我们接受的误差……；

步骤4，基于分割网络模型分割同类图像集中的其余图像：将同类图像集中的其余n-1张图像和由步骤2获得的变换掩码图进行结合，构成4通道测试图像并输入到分割网络模型中，在上采样层图像恢复到原始输入图像尺寸，从而对每个像素都产生了一个预测,同时保留了原始输入图像中的空间信息,最后在上采样层的特征图上进行逐像素分类，从而得到n-1张测试图像的预测分割掩码；

步骤5，再分割：检查步骤4所得到的分割结果：

如果所有图像均已获得满意的分割结果，则分割结束；这里我们定义一个差值diff：

公式1中，分母480*480是图像的所有像素点的个数，因为我们的图像集中的图像的大小是480*480；

所述满意的分割结果是指，这些图像的预测分割掩码的diff值都处在-15％至15％之间；

否则，选择一副分割效果较差的图像作为新的分割样本，并返回步骤1训练新的分割网络模型，实现对其余分割欠佳图像的分割优化，并重复该过程直至所有图像分割完成；所述分割效果较差图像是指这些图像的预测分割掩码的diff值都不在-15％至15％之间。