[发明专利]一种基于深度学习的超像素分割方法

权利要求书

1.一种基于深度学习的超像素分割方法，其特征在于包含以下步骤：

S1:网络模型初始化：首先选用U型编解码网络结构作为初始网络，所述的U型编解码网络结构由编码器和解码器两部分构成，并且在编码器和解码器相同尺度位置处包括跨层连接；

S2：网络模型过参数化：设计多分支特征提取模块，对初始网络编码器中任意层数的特征提取层进行替换，实现网络模型过参数化；其中，所述的多分支特征提取模块为并行的多分支结构，该模块的输入与输出不要求为相同尺寸，其中不同分支由不同结构的卷积神经网络构成，每一分支的输入彼此相同，每一分支的输出尺寸彼此相同，并且每一分支具备一个初始化的可学习权重，不同分支的输出通过各自分支的可学习权重进行逐元素相加，后接批归一化和激活函数，进而构成并行的多分支特征提取结构；

S3：网络剪枝：对通过利用多分支特征提取模块替换初始网络编码器中任意层数的特征提取层得到的过参数化网络进行训练，至损失函数曲线趋于平缓时停止训练，利用网络剪枝策略对训练完成的网络模型完成过参数化网络剪枝，得到针对超像素分割任务的编码器；其中，所述的网络剪枝策略为针对多分支特征提取模块中的每一分支，当该分支训练得到的对应的可学习权重小于设定的阈值时，剪去该分支，反之保留该分支，当所有分支权重均小于阈值时，仅保留权值最大的分支；

S4：在网络剪枝之后的解码器结构上添加跨层特征融合，实现由上至下，再由下至上的解码结构，其强化后的解码器简称为TDT(Top-down-top)解码器。所述的跨层特征融合位于U型编解码网络由上至下解码阶段之后，用于实现由微至广的跨尺度特征融合，进而提升原尺度特征图质量，其跨尺度特征融合过程为由下至上的解码结构。所述的跨尺度特征融合具体操作为：在U型编解码网络原有框架下，将解码器前端大感受野下的任意一层的输出特征进行上采样后，与解码器后端原尺度网络层的输出特征进行拼接并后接卷积操作，实现两层间跨尺度的信息融合，实现网络解码强化，并后接卷积和Softmax激活函数，进而实现基于概率形式的超像素分割结果计算。通过串联U型编解码网络由上至下的解码结构和跨尺度特征融合由下至上的解码结构，实现了由上至下，再由下至上的TDT解码器整体解码结构的构建；

S5：对解码强化后的网络进行训练，并利用训练完成的网络对图像进行超像素分割。

2.根据权利要求1所述的一种基于深度学习的超像素分割方法，其特征在于：步骤S3中所述阈值λ的取值范围为[0.85w,0.95w]，其中w为初始化权重。

3.根据权利要求1所述的一种基于深度学习的超像素分割方法，其特征在于：所述的U型编解码网络为SpixelFCN编解码网络。