一种基于提前终止的图像识别推理加速方法

基本信息

申请号 CN201910457924.7 申请日 -
公开(公告)号 CN110163295A 公开(公告)日 2019-08-23
申请公布号 CN110163295A 申请公布日 2019-08-23
分类号 G06K9/62;G06N3/04;G06N5/04 分类 计算;推算;计数;
发明人 高科;曾加贝;张弛 申请(专利权)人 南京沛亭数字科技有限公司
代理机构 北京金智普华知识产权代理有限公司 代理人 四川智盈科技有限公司
地址 610051 四川省成都市成华区迎晖路138号附1号
法律状态 -

摘要

摘要 本发明涉及图像识别技术领域,提供了一种基于提前终止的图像识别推理加速方法,在训练阶段对于N分类任务,在中间层之后加一个基于归一化指数函数的N+1分类交叉熵损失函数,前N个结点对应N分类中每个分类的打分,第N+1类得分用于判断图像识别的可靠性;在推理阶段,给定测定图像,依次通过每一层,遇到待考察的层,先根据该待考察的层的输出来计算用于做图像识别的可靠程度;若可靠程度不满足条件,则进入下一层;若可靠程度满足条件,则根据该待考察的层的输出进行图像分类判断,输出判断结果,终止推理过程。本发明加速效果明显,可加速包括物体识别、场景识别、人脸识别、表情识别、性别判断等在内的二分类或者多分类图像识别的推理。