一種基于提前終止的圖像識別推理加速方法

基本信息

申請?zhí)?/td> CN201910457924.7 申請日 -
公開(公告)號 CN110163295A 公開(公告)日 2019-08-23
申請公布號 CN110163295A 申請公布日 2019-08-23
分類號 G06K9/62;G06N3/04;G06N5/04 分類 計算;推算;計數;
發(fā)明人 高科;曾加貝;張弛 申請(專利權)人 南京沛亭數字科技有限公司
代理機構 北京金智普華知識產權代理有限公司 代理人 四川智盈科技有限公司
地址 610051 四川省成都市成華區(qū)迎暉路138號附1號
法律狀態(tài) -

摘要

摘要 本發(fā)明涉及圖像識別技術領域,提供了一種基于提前終止的圖像識別推理加速方法,在訓練階段對于N分類任務,在中間層之后加一個基于歸一化指數函數的N+1分類交叉熵損失函數,前N個結點對應N分類中每個分類的打分,第N+1類得分用于判斷圖像識別的可靠性;在推理階段,給定測定圖像,依次通過每一層,遇到待考察的層,先根據該待考察的層的輸出來計算用于做圖像識別的可靠程度;若可靠程度不滿足條件,則進入下一層;若可靠程度滿足條件,則根據該待考察的層的輸出進行圖像分類判斷,輸出判斷結果,終止推理過程。本發(fā)明加速效果明顯,可加速包括物體識別、場景識別、人臉識別、表情識別、性別判斷等在內的二分類或者多分類圖像識別的推理。