全文摘要
本发明公开了一种基于深度神经网络模型的图像识别方法,该方法包括:获取待识别的目标图像;将目标图像输入至利用通道的表征能力对深度神经网络模型进行通道剪枝后获得的目标模型中;利用目标模型中的辅助分类器对目标图像进行分类处理,获得识别结果。由于识别目标图像的目标模型是基于通道的表征能力进行剪枝后的模型,因而,对目标图像进行识别时的计算量可大幅度降低。本发明还公开了一种基于深度神经网络模型的图像识别装置、设备及可读存储介质,具有相应的技术效果。
主设计要求
1.一种基于深度神经网络模型的图像识别方法,其特征在于,包括:获取待识别的目标图像;将所述目标图像输入至利用通道的表征能力对深度神经网络模型进行通道剪枝后获得的目标模型中;利用所述目标模型中的辅助分类器对所述目标图像进行分类处理,获得识别结果;其中,获得所述目标模型的过程,包括:在所述深度神经网络模型中插入批量归一化层、线性整流层、平均池化层构建辅助分类器;在所述深度神经网络模型中插入辅助损失函数,并与所述深度神经网络模型的重构损失函数构成目标损失函数;利用所述目标损失函数结合通道的表征能力,对所述辅助分类器进行训练,获得目标模型。
设计方案
1.一种基于深度神经网络模型的图像识别方法,其特征在于,包括:
获取待识别的目标图像;
将所述目标图像输入至利用通道的表征能力对深度神经网络模型进行通道剪枝后获 得的目标模型中;
利用所述目标模型中的辅助分类器对所述目标图像进行分类处理,获得识别结果;
其中,获得所述目标模型的过程,包括:
在所述深度神经网络模型中插入批量归一化层、线性整流层、平均池化层构建辅助分 类器;
在所述深度神经网络模型中插入辅助损失函数,并与所述深度神经网络模型的重构损 失函数构成目标损失函数;
利用所述目标损失函数结合通道的表征能力,对所述辅助分类器进行训练,获得目标 模型。
2.根据权利要求1所述的基于深度神经网络模型的图像识别方法,其特征在于,在所述 深度神经网络模型中插入辅助损失函数,包括:
在所述深度神经网络模型中插入交叉熵损失函数。
3.根据权利要求2所述的基于深度神经网络模型的图像识别方法,其特征在于,所述利 用所述目标损失函数结合通道的表征能力,对所述辅助分类器进行训练,获得目标模型,包 括:
利用所述目标损失函数并结合通道的表征能力,在所述深度神经网络模型中选择待剪 枝的冗余通道;
在所述深度神经网络模型中,剪去所述冗余通道,获得目标模型。
4.根据权利要求3所述的基于深度神经网络模型的图像识别方法,其特征在于,所述利 用所述目标损失函数并结合通道的表征能力,在所述深度神经网络模型中选择待剪枝的冗 余通道,包括:
获取表征所述深度神经网络模型中各个通道的重要性的通道选择向量;
利用所述目标损失函数对所述通道选择向量和模型参数进行优化;
将优化后通道选择向量的向量元素为0的通道确定为冗余通道。
5.根据权利要求4所述的基于深度神经网络模型的图像识别方法,其特征在于,所述利 用所述目标损失函数对所述通道选择向量和模型参数进行优化,包括:
随机选择训练样本,并利用随机梯度下降算法和贪婪算法,令目标损失函数 设计说明书
技术领域
本发明涉及智能识别技术领域,特别是涉及一种基于深度神经网络模型的图像识 别方法、装置、设备及可读存储介质。
背景技术
目前,图像分类、人脸识别等图像分类识别中,深度神经网络模型至关重要。然而, 深度神经网络具有参数多、计算量大的特点,造成了大量的内存需求以及计算负担,使得深 度神经网络模型难以应用到存储和计算资源有限的如手机等硬件设备中。
综上所述,如何有效地解决减少图像识别的计算量等问题,是目前本领域技术人 员急需解决的技术问题。
发明内容
本发明的目的是提供一种基于深度神经网络模型的图像识别方法、装置、设备及 可读存储介质,以实现减少基于深度神经网络模型的图像识别时的计算量。
为解决上述技术问题,本发明提供如下技术方案:
一种基于深度神经网络模型的图像识别方法,包括:
获取待识别的目标图像;
将所述目标图像输入至利用通道的表征能力对深度神经网络模型进行通道剪枝 后获得的目标模型中;
利用所述目标模型中的辅助分类器对所述目标图像进行分类处理,获得识别结 果;
其中,获得所述目标模型的过程,包括:
在所述深度神经网络模型中插入批量归一化层、线性整流层、平均池化层构建辅 助分类器;
在所述深度神经网络模型中插入辅助损失函数,并与所述深度神经网络模型的重 构损失函数构成目标损失函数;
利用所述目标损失函数结合通道的表征能力,对所述辅助分类器进行训练,获得 目标模型。
优选地,在所述深度神经网络模型中插入辅助损失函数,包括:
在所述深度神经网络模型中插入交叉熵损失函数。
优选地,所述利用所述目标损失函数结合通道的表征能力,对所述辅助分类器进 行训练,获得目标模型,包括:
利用所述目标损失函数并结合通道的表征能力,在所述深度神经网络模型中选择 待剪枝的冗余通道;
在所述深度神经网络模型中,剪去所述冗余通道,获得目标模型。
优选地,所述利用所述目标损失函数并结合通道的表征能力,在所述深度神经网 络模型中选择待剪枝的冗余通道,包括:
获取表征所述深度神经网络模型中各个通道的重要性的通道选择向量;
利用所述目标损失函数对所述通道选择向量和模型参数进行优化;
将优化后通道选择向量的向量元素为0的通道确定为冗余通道。
优选地,所述利用所述目标损失函数对所述重要通道的通道选择向量和模型参数 进行优化,包括:
随机选择训练样本,并利用随机梯度下降算法和贪婪算法,令目标损失函数 设计图
相关信息详情
申请码:申请号:CN201910004752.8
申请日:2019-01-03
公开号:CN109344921A
公开日:2019-02-15
国家:CN
国家/省市:43(湖南)
授权编号:CN109344921B
授权时间:20190423
主分类号:G06K9/62
专利分类号:G06K9/62;G06N3/04
范畴分类:40B;
申请人:湖南极点智能科技有限公司
第一申请人:湖南极点智能科技有限公司
申请人地址:410000 湖南省长沙市岳麓区潇湘中路328号58众创508-2
发明人:谭明奎;吴希贤
第一发明人:谭明奎
当前权利人:湖南极点智能科技有限公司
代理人:罗满
代理机构:11227
代理机构编号:北京集佳知识产权代理有限公司
优先权:关键词:当前状态:审核中
类型名称:外观设计