一种基于提前终止的图像识别推理加速方法
基本信息
申请号 | 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类得分用于判断图像识别的可靠性;在推理阶段,给定测定图像,依次通过每一层,遇到待考察的层,先根据该待考察的层的输出来计算用于做图像识别的可靠程度;若可靠程度不满足条件,则进入下一层;若可靠程度满足条件,则根据该待考察的层的输出进行图像分类判断,输出判断结果,终止推理过程。本发明加速效果明显,可加速包括物体识别、场景识别、人脸识别、表情识别、性别判断等在内的二分类或者多分类图像识别的推理。 |
