[发明专利]一种改进重构网络的胶囊网络图像分类识别方法

摘要

本发明公开一种改进重构网络的胶囊网络图像分类识别方法：S1，构建胶囊网络；S2，输入图像训练集合至所述胶囊网络，所述胶囊网络经训练学习后完成图像分类识别校准；S3，输入待分类图像至所述胶囊网络，所述工作网络的输出向量vj中数值最大的一个即为得到的识别结果；S4，所述胶囊网络输出所述待分类图像的识别结果；其中胶囊网络的重构网络结构为反卷积操作。有益效果：提出了一种新的重构网络结构，通过反卷积操作把向量还原为图像，对比还原的图像和原始图像的误差来调节网络参数，减小了计算参数量，为硬件设备空出了更多运行内存。

主权项

1.一种改进重构网络的胶囊网络图像分类识别方法，其特征在于：S1，构建胶囊网络，所述胶囊网络设置有工作网络与校对网络，所述工作网络用于输入图像并输出该图像的识别结果，所述校对网络用于训练调节工作网络参数；所述工作网络包括卷积结构和全连接结构，所述卷积结构的卷积输出端连接全连接结构的全连接输入端，所述卷积结构为依次连接的卷积层和PrimaryCaps层，所述全连接结构为依次进行权重计算、动态路由调节、激活函数运算的网络结构；所述校对网络包括并行的margin loss运算结构和重构网络结构，所述margin loss运算结构的损失输入端连接所述全连接结构的全连接输出端，所述重构网络结构的重构输入端分别连接所述全连接结构的全连接输出端和输入图像的向量层，所述margin loss运算结构的损失输出端与所述重构网络结构的重构输出端分别连接Loss层的损失函数输入端，所述Loss层的损失函数输出端连接优化函数计算层；所述重构网络结构包括依次连接的Reshape层、反卷积结构、Flatten层、和方差计算层，所述和方差计算层的方差输入端分别连接Flatten层和输入图像的向量层，所述和方差计算层的方差输出端连接Loss层的损失函数输入端；S2，输入图像训练集合至所述胶囊网络，所述胶囊网络经训练学习后完成图像分类识别校准；S3，输入待分类图像至所述胶囊网络，所述工作网络的输出向量vj中数值最大的一个即为得到的识别结果；S4，所述胶囊网络输出所述待分类图像的识别结果。