一种利用共基胶囊投影进行图像分类的方法

摘要

本发明属于图像分类技术领域，公开了一种利用共基胶囊投影进行图像分类的方法，包括下述步骤：(1)利用多层卷积网络提取输入图像的特征，获得特征图；(2)将特征图映射成一个一维的特征向量X；(3)对特征向量X进行特征变换，将特征向量X分为N组，并组合向量为特征矩阵；(4)将特征矩阵进行共基胶囊投影，投影到多个胶囊子空间，计算每个子空间投影后的向量模长和，根据模长和的大小进行图像分类预测。本发明利用共基胶囊投影思想将其特征投影到多个胶囊子空间，然后再进行图像分类任务的预测，经过实验发现该网络对大小规模的图像都能够适应，并且即使采用较小的数据集训练也能达到非常好的分类效果。