本发明涉及图像分类技术领域,具体公开了一种基于残差胶囊神经网络的图像分类系统,搭载有残差胶囊神经网络,该网络包括第一、第二、第三残差胶囊模块,数字胶囊层;第一残差胶囊模块包括第一残差卷积子网和第一初始胶囊层,第二残差胶囊模块包括第二残差卷积子网和第二初始胶囊层,第三残差胶囊模块包括第三残差卷积子网和第三初始胶囊层。本发明基于残差学习的思想,通过多级残差胶囊模块串联的方式实现了多级信息复用,解决了网络加深之后梯度消失的问题,并且引入了空洞卷积,有助于提取更多的特征;运用转置卷积构建重构网络,在降低网络参数量的同时提升了网络性能;基于忆阻交叉阵列实现胶囊网络的硬件部署,以解决计算量大的问题。
📄 2021115871005
📂 G06V10_764
👤 西南大学
📅 2021-12-23