[发明专利]基于混合深度可分离膨胀卷积的图像分类模型及其分类方法

说明书

本发明公开了一种基于混合深度可分离膨胀卷积的图像分类模型，构建过程包括：从前至后将深度可分离膨胀卷积层、特征连接层、卷积层、批标准化层和修正线性单元层封装成混合深度可分离膨胀卷积模块；从前至后将卷积层、批量归一化层、修正线性单元层、混合深度可分离膨胀卷积模块、最大值池化层、压平层、随机失活层和全连接层封装构成深度神经网络的主干网络；对主干网络的参数权重进行随机初始化，并预设迭代次数和批量归一化层的动量参数；采用随机梯度下降法优化网络模型的参数，重复迭代计算，直至损失值收敛，得到最优的网络模型。通过上述方案，本发明具有架构简单、计算工作量少、分类精准等优点。

技术领域

本发明涉及图像处理技术领域，尤其是基于混合深度可分离膨胀卷积的图像分类模型及其分类方法。

背景技术

图像是可以客观地展示自然事物，是人们认知世界的重要资源，而且技术人员可以通过分析图像获得有益的信息，发展相关算法。图像分类属于计算机视觉方向，在医疗、食品安全等领域广泛应用。

目前，现有技术中的图像分类算法主要思想就是给需要分类的图像集合分配对应的标签，对于计算机来说，一张图像就是一个像素矩阵，其利用相关算法技术提取像素矩阵中有效的信息，这与人对图像的认知方式不同。传统图像分类算法主要分为手工特征提取、特征编码、分类器分类这三步骤；其中，常用的手工特征有局部二值化模式特征(LocalBinary Pattern，LBP)、尺度不变特征变换(Scale-Invariant Feature Transform，SIFT)以及方向梯度直方图特征(Histogram of Oriented Gradient，HOG)等。特征编码是为了去除冗余的特征信息，提高表征信息的鲁棒性。通用的分类器有支持向量机(Support VectorMachine，SVM)和自适应增强分类器(Adaptive Boosting，AdaBoost)等。由于手工特征泛化性差，从而影响到传统图像分类算法在实际场景中的应用，在精度和速度上都没有多大的优势。随着计算机算力的提升，深度学习算法得到了迅速的发展，深度学习的图像分类算法也因此开始崛起。

2012年Alex Krizhevsky等人提出了Alexnet网络结构，对ImageNet数据集进行学习，取得了2012年ImageNet比赛的冠军，从此之后不断出现深度学习算法应用到大数据图像分类上。深度学习的图像分类算法通过利用卷积神经网络提取样本的卷积特征，具有更加丰富的语义信息，能让算法模型学习到更加高效的表达能力，与手工特征相比，更为精准。但是，深度学习算法的图像分类算法在训练网络模型时，为了减少特征维度，都会添加特征下采样操作,一定程度上避免网络出现过拟合现象，如池化层，这样会造成卷积特征丢失细节信息。Jonathan Long等人提出全卷积网络结构，其中，使用反卷积层增大特征图的分辨率，一定程度上恢复些细节信息但较为粗糙。同年，Kaiming He等人提出残差深度神经网络，采用跳跃连接结构将深层网络层特征和浅层网络层特征融合，提升特征图中细节信息的使用率。

现有技术中的深度学习的图像分类算法中采用池化层或步长为2的卷积层对特征图像进行下采样操作，减少数量，扩大感受野，增加特征图的语义信息，但是，其会丢失特征图的细节信息。

综上所述，上述方法都只能一定程度上修补特征图的细节信息，但是会增加网络模型的参数量，使网络模型变得复杂，难优化，所以特征图细节信息的丢失问题还是影响着分类检测精度。

另外，在专利申请号为“201910818758.9”、名称为“基于轻量级深度神经网络的工业产品缺陷图像分类方法”的中国发明专利中，其包含：1、工业产品图像数据集的准备；2、搭建轻量级深度神经网络；3、将工业产品缺陷图像数据集输入到搭建好的轻量级深度神经网络中，经过网络训练，提取偏光片图像多尺度特征，将提取到的特征输入Softmax层进行分类，得到分类模型；4、将测试图像输入分类模型中，将图像属于某一类别的概率和图像对应的标签输入到Accuracy层，输出图像的正确分类结果。但是，该技术存在以下缺陷：