首页/ 华平信息技术股份有限公司/ 专利详情

基于梯度维持的图像分类方法、存储介质及电子设备

基本信息

申请号 CN202011377685.3 申请日 -
公开（公告）号 CN112348118A 公开（公告）日 2021-02-09
申请公布号 CN112348118A 申请公布日 2021-02-09
分类号 G06K9/62(2006.01)I;G06N3/04(2006.01)I 分类 计算；推算；计数;
发明人 何良华;任强;施小春;李鹏飞;罗涛 申请（专利权）人 华平信息技术股份有限公司
代理机构 上海光华专利事务所（普通合伙） 代理人 徐秋平
地址 200433上海市杨浦区国权北路1688弄69号
法律状态 -

摘要

摘要 本发明提供一种基于梯度维持的图像分类方法、存储介质及电子设备，所述基于梯度维持的图像分类方法包括：获取待训练的图像集；将所述图像集划分为训练集和测试集；利用所述训练集训练分类胶囊网络模型，且在训练过程中通过图像预测胶囊的路由梯度系数维持胶囊层叠加时的梯度稳定；将所述测试集输入所述分类胶囊网络模型中进行图像分类的性能测试。本发明在图像分类所用的胶囊网络模型中通过路由梯度系数维持梯度稳定。不仅在某种程度上抑制了梯度的消失，而且还可以适当的可以放大梯度并将梯度更平滑地扩展到模型前面的网络层，提高了胶囊网络模型的图像分类性能。