1.一种基于密集连接解码网络的轮廓检测方法,其特征在于包括以下步骤:
A、构建深度神经网络结构,深度神经网络结构具体如下:
预设初始等级、融合等级、最终等级,其中初始等级及最终等级均设有1个,融合等级设有1-5个;
其中,初始等级中包含5个阶段,分别为阶段Ⅰ-阶段Ⅴ,每个阶段中分别设置有一个以上的卷积层;
所述的阶段Ⅰ的第一卷积层的输入响应为原始图像,阶段Ⅰ的其他卷积层的输入响应为该阶段上一卷积层的输出响应;阶段Ⅱ-阶段Ⅴ中,除了该阶段中的第一卷积层的输入响应之外,该阶段中的其他卷积层的输入响应为上一卷积层的输出响应;阶段Ⅰ-阶段Ⅳ中最后一个卷积层的输出响应,一方面经过最大池化后作为下一个阶段的第一卷积层的输入响应;另一方面,作为输入响应输入到第一个融合等级中;
所述的融合等级中设有1-5个融合层组,各个融合层组包含数量与融合等级数量相同的融合层,同一融合层组中的各个融合层分别与各个融合等级相对应,各个融合层预设有与其相对应的结果图通道数量以及结果图尺寸,同一融合层组中的各个融合层所对应的结果图通道数量以及结果图尺寸相同;
对于各个融合等级中的各个融合层:分别将来自初始等级或上一融合等级的各输入响应中的结果图通道数量以及结果图尺寸调整为该融合层所对应的结果图通道数量以及结果图尺寸,分别得到该融合层的多个预处理响应,其中采用卷积后激活的方式对结果图通道数量进行调整;依次对各预处理响应同一通道的结果图中的各像素点的结果值求均值,作为该通道的各像素点的融合值,从而得到各通道的融合结果图,对各通道的融合结果图的各像素点的融合值进行归一化处理后,作为各通道的融合结果图的各像素点的归一化融合值,并将其作为下一融合等级或最终等级的输入响应;
最终等级包含最终融合层,最终融合层将来自最后一个融合等级的各个输入响应的结果图通道数量调整为单通道,并将来自最后一个融合等级的各个输入响应的结果图尺寸调整为原始图像的尺寸,分别得到最终融合层的多个预处理响应,对各预处理响应的结果图中的各像素点的结果值求均值,作为各像素点的最终融合值,对各像素点的最终融合值进行归一化处理后,得到各像素点的归一化最终融合值,作为各像素点的轮廓响应值,即得到深度神经网络的输出响应;
所述的阶段Ⅰ-阶段Ⅴ以及各融合层涉及的卷积表达式为m*n-k conv+relu,其中,m*n表示卷积核的大小,k表示输出通道数,conv表示卷积公式,relu表示激活函数;m*n、k均为预设值;所述的最终融合层的卷积表达式为m*n-k conv;
B、预设训练数据集,训练数据集包含训练图像及其对应的真实轮廓图,输入训练数据集中的训练图像及其真实轮廓图,预设图像批量大小,权重衰减参数,学习率,学习率衰减,最大迭代次数,优化方案和损失函数,以训练数据集中的训练图像作为输入,对应的真实轮廓图作为理想化输出,训练深度神经网络,调整各个卷积核的参数使得深度神经网络输出响应最接近理想化输出,得到训练完毕的深度神经网络;
C、利用双线性插值算法对待检测图像进行处理,分别得到尺寸是待检测图像2倍的待检测放大图像、尺寸是待检测图像0.5倍的待检测缩小图像,将待检测图像、待检测放大图像、待检测缩小图像分别作为深度神经网络的输入,经过深度神经网络输出分别得到待检测图像轮廓响应图、待检测放大图像轮廓响应图、待检测缩小图像轮廓响应图,将各像素点在上述三个轮廓响应图中对应的特征值取平均值,作为该像素点最终轮廓响应特征值,从而得到各像素点最终轮廓响应特征值;
D、对各像素点的最终轮廓响应特征值使用非极大值抑制和二值化处理,得到各像素点的最终轮廓值,进而得到最终轮廓图。
2.如权利要求1所述的基于密集连接解码网络的轮廓检测方法,其特征在于:所述的步骤B中的损失函数具体为:
LT=L+βLS (1);
L=-αlogpi·|L+|-(1-α)log(1-pi)·|L-| (2);
其中L+、L-分别表示训练图像的真实轮廓图的边缘像素集及非边缘像素集,i为边缘像素集中的像素点编号,pi为编号为i的像素点的深度神经网络的轮廓响应值;α=|L-|/|L|,|L+|、|L-|分别为边缘像素集中像素点个数及非边缘像素集中像素点个数,|L|为|L+|、|L-|之和;
LS为融合层损失总和,通过对每个融合层的输出响应利用公式(2)进行损失计算后求和得到,β为比例参数。
3.如权利要求2所述的基于密集连接解码网络的轮廓检测方法,其特征在于:卷积均为带有偏置的卷积,偏置的个数与当前的卷积核的个数相对应;
卷积conv的公式如下:
outputj=inputj*kernelj+biasj (3);
其中outputj为当前阶段卷积的第j个输出响应,inputj为当前阶段卷积的第j个输入响应,kernelj为第j个卷积核,biasj为第j个偏置,*为卷积符号,j=1,2,...,k,k为当前阶段卷积核的个数。
4.如权利要求3所述的基于密集连接解码网络的轮廓检测方法,其特征在于:所述的激活函数为:
relu(x)=max(0,x) (4)。
5.如权利要求4所述的基于密集连接解码网络的轮廓检测方法,其特征在于:最初等级涉及的卷积层个数分别为2个、2个、3个、3个、3个;
最初等级涉及的输出通道数分别为64个、128个、256个、512个、512个;
最初等级、融合等级中涉及的的卷积核大小为3*3像素平方。
6.如权利要求5所述的基于密集连接解码网络的轮廓检测方法,其特征在于:对结果图尺寸调整的方法为池化或者双线性插值,其中池化对应于将结果图尺寸调整变小,双线性插值对应于将结果图尺寸调整变大;
所述的归一化处理为将各像素点的结果值与大小为1*1像素平方的卷积核进行卷积,并且该卷积核函数为sigmoid函数。
7.如权利要求6所述的基于密集连接解码网络的轮廓检测方法,其特征在于:所述的预设图像批量大小表示每一批次训练输入的图像数量,其范围为8-32幅,权重衰减范围为(1e-4)-(2e-4),学习率的范围为(1e-5)-(1e-9),学习率衰减的范围为0-0.1;
最大迭代次数的范围为5000-60000次,优化方案为动量为0.9的随机梯度下降法。
8.如权利要求7所述的基于密集连接解码网络的轮廓检测方法,其特征在于:所述的最大池化为2*2最大池化。