【技术实现步骤摘要】
基于自适应空间模式的深度胶囊网络图像分类方法及系统
本专利技术涉及图像分类领域,尤其涉及一种基于自适应空间模式的深度胶囊网络图像分类方法及系统。
技术介绍
传统的深度学习图像分类算法包括:堆叠自编码网络、深度置信神经网络,递归神经网络,生成对抗性神经网络以及卷积神经网络等,其中传统的基于神经网络的图像分类方法通过卷积、池化、全连接层等步骤进行图像特征提取。然而,该类方法往往网络模型训练需要大量的数据样本,且训练过程复杂,复杂的场景特征下图像关键兴趣点特征提取困难。且对于传统的神经网络方法,其卷积核中采样点的位置通常是固定不变的,其不能根据真实物体的实际空间形态分布自适应的调整学习区域。因此,当数据集结构复杂的情况下,现有的深度学习方法往往无法学习到清晰轮廓特征的图像边缘信息,因而面临无法精准高效地识别图像复杂纹理信息的问题。
技术实现思路
为了解决上述技术问题,本专利技术的目的是提供一种基于自适应空间模式的深度胶囊网络图像分类方法及系统,解决空间结构特征复杂的高光谱图像分类数据集分类精度不准确的问题。r>本专利技术所采用本文档来自技高网...
【技术保护点】
1.基于自适应空间模式的深度胶囊网络图像分类方法,其特征在于,包括以下步骤:/n对训练用图像进行图像预处理,并构建训练集;/n基于训练集进行适应空间单元深度胶囊网络的训练,得到训练完成的深度胶囊网络;/n获取待测图像并将待测图像输入训练完成的深度胶囊网络进行图像分类,得到图像分类结果;/n所述深度胶囊网络包括ASPConvs模块、ASPCaps模块和全连接胶囊层。/n
【技术特征摘要】
1.基于自适应空间模式的深度胶囊网络图像分类方法,其特征在于,包括以下步骤:
对训练用图像进行图像预处理,并构建训练集;
基于训练集进行适应空间单元深度胶囊网络的训练,得到训练完成的深度胶囊网络;
获取待测图像并将待测图像输入训练完成的深度胶囊网络进行图像分类,得到图像分类结果;
所述深度胶囊网络包括ASPConvs模块、ASPCaps模块和全连接胶囊层。
2.根据权利要求1所述一种基于自适应空间模式的深度胶囊网络图像分类方法,其特征在于,所述对训练用图像进行图像预处理,并构建训练集这一步骤,其具体包括:
获取训练用的高光谱图像X∈RH×W×D;
基于主成分分析算法将数据降到d维,得到图像训练块Yk∈Rm×m×d;
结合图像训练块和对应的标签构建训练集
3.根据权利要求2所述一种基于自适应空间模式的深度胶囊网络图像分类方法,其特征在于,所述基于训练集进行适应空间单元深度胶囊网络的训练,得到训练完成的深度胶囊网络这一步骤,其具体包括:
将训练集输入到适应空间单元深度胶囊网络;
基于ASPConvs模块对训练图像块进行卷积处理,提取得到次高级目标图像特征;
基于ASPCaps模块对次高级图像特征进行胶囊分组和预测处理,重新提取得到高级语义特征;
基于全连接胶囊网络层对高级语义特征进行概率分类,得到分类结果。
4.根据权利要求3所述一种基于自适应空间模式的深度胶囊网络图像分类方法,其特征在于,所述基于ASPConvs模块对训练图像块进行卷积处理,提取得到次高级目标图像特征这一步骤,其具体包括:
给定训练图像块Yk,引入空洞卷积规则并自适应学习寻找采样位置;
基于采样位置提取得到次高级目标图像特征。
5.根据权利要求4所述一种...
【专利技术属性】
技术研发人员:王锦萍,李军,谭晓军,黄力,陈霞,
申请(专利权)人:中山大学,惠州市德赛西威汽车电子股份有限公司,
类型:发明
国别省市:广东;44
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。