基于胶囊网络的图像分类方法、存储介质及电子设备

基本信息

申请号 CN202011377744.7 申请日 -
公开(公告)号 CN112348119A 公开(公告)日 2021-02-09
申请公布号 CN112348119A 申请公布日 2021-02-09
分类号 G06K9/62(2006.01)I 分类 计算;推算;计数;
发明人 何良华;任强;高世忠;孙振宇;王军 申请(专利权)人 华平信息技术股份有限公司
代理机构 上海光华专利事务所(普通合伙) 代理人 徐秋平
地址 200433上海市杨浦区国权北路1688弄69号
法律状态 -

摘要

摘要 本发明提供一种基于胶囊网络的图像分类方法、存储介质及电子设备,所述基于胶囊网络的图像分类方法包括:获取待训练的图像集;将所述图像集划分为训练集和测试集;利用所述训练集训练分类胶囊网络模型,在训练过程中根据所述训练集中图像特征仿射变换后形成的图像预测胶囊进行路由迭代;将所述测试集输入所述分类胶囊网络模型中进行图像分类的性能测试。本发明提出了不涉及耦合系数参与训练的自适应路由方法,在路由的迭代过程中可以使胶囊自适应地改变自身的特征,使胶囊向更有判别性的方向偏移,避免在出现梯度消失问题的同时也提高了模型的性能。