[發明專利]用于識別動物的方法和裝置

有效

• 申請號: 201810043970.8
• 申請日: 2018-01-17
• 公開（公告）號: CN108171274B
• 公開（公告）日: 2019-08-09
• 發明（設計）人: 袁宇辰;周峰;張成月
• 申請（專利權）人: 百度在線網絡技術（北京）有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京英賽嘉華知識產權代理有限責任公司 11204
• 代理人: 王達佐;馬曉亞

本申請實施例公開了用于識別動物的方法和裝置。該方法的一具體實施方式包括：獲取待識別圖像；將待識別圖像輸入至預先訓練的第一識別模型，得到與待識別圖像對應的第一識別結果；基于第一識別結果，確定待識別圖像中是否存在動物；響應于確定待識別圖像中存在動物，將待識別圖像輸入至預先訓練的第二識別模型，得到與待識別圖像對應的第二識別結果；將第一識別結果和第二識別結果進行融合，生成與待識別圖像對應的第三識別結果。該實施方式將利用第一識別模型得到的第一識別結果和利用第二識別模型得到的第二識別結果相結合，從而提高了動物識別的準確度。

技術領域

本申請實施例涉及計算機技術領域，具體涉及圖像識別技術領域，尤其涉及用于識別動物的方法和裝置。

背景技術

世界上的動物類別繁多，人們通過肉眼通常只能辨別出極少數類別的動物。因此，幫助用戶進行動物識別成為了一種需求。現有的動物識別方式通常是通過相似圖進行檢索查詢來識別圖像中的動物的類別。

發明內容

本申請實施例提出了用于識別動物的方法和裝置。

第一方面，本申請實施例提供了一種用于識別動物的方法，該方法包括：獲取待識別圖像；將待識別圖像輸入至預先訓練的第一識別模型，得到與待識別圖像對應的第一識別結果；基于第一識別結果，確定待識別圖像中是否存在動物；響應于確定待識別圖像中存在動物，將待識別圖像輸入至預先訓練的第二識別模型，得到與待識別圖像對應的第二識別結果；將第一識別結果和第二識別結果進行融合，生成與待識別圖像對應的第三識別結果。

在一些實施例中，第一識別結果包括待識別圖像中存在預設類別集合中的每種類別的動物的第一置信度和不存在動物的第一置信度；以及基于第一識別結果，確定待識別圖像中是否存在動物，包括：對第一識別結果中的待識別圖像中存在預設類別集合中的每種類別的動物的第一置信度進行加權求和，得到待識別圖像中存在動物的第一置信度；基于待識別圖像中存在動物的第一置信度，確定待識別圖像中是否存在動物。

在一些實施例中，第二識別結果包括待識別圖像中存在預設類別集合中的每種類別的動物的第二置信度，第三識別結果包括待識別圖像中存在預設類別集合中的每種類別的動物的第三置信度；以及將第一識別結果和第二識別結果進行融合，生成與待識別圖像對應的第三識別結果，包括：對于預設類別集合中的每種類別，對第一識別結果中的待識別圖像中存在該種類別的動物的第一置信度與第二識別結果中的待識別圖像中存在該種類別的動物的第二置信度進行加權求和；對所得到的和進行歸一化處理，生成待識別圖像中存在該種類別的動物的第三置信度。

在一些實施例中，該方法還包括：按照數值大小順序對第三識別結果中的待識別圖像中存在預設類別集合中的每種類別的動物的第三置信度進行排序；從數值大的一側開始選取出預設數目個第三置信度，輸出所選取出的第三置信度和與所選取出的第三置信度對應的類別。

在一些實施例中，該方法還包括：將第三識別結果中的待識別圖像中存在預設類別集合中的每種類別的動物的第三置信度與預設閾值進行比較；若存在大于預設閾值的第三置信度，輸出大于預設閾值的第三置信度和與大于預設閾值的第三置信度對應的類別。

在一些實施例中，該方法還包括：響應于確定待識別圖像中不存在動物，輸出用于提示待識別圖像中不存在動物的提示信息和待識別圖像中不存在動物的第一置信度。

在一些實施例中，第一識別模型和第二識別模型通過如下步驟訓練得到：獲取樣本圖像集合和與樣本圖像集合中的樣本圖像對應的標注結果；利用機器學習方法，基于樣本圖像集合和與樣本圖像集合中的樣本圖像對應的標注結果對預設的第一卷積神經網絡進行訓練，得到第一識別模型；利用機器學習方法，基于樣本圖像集合和與樣本圖像集合中的樣本圖像對應的標注結果對預設的第二卷積神經網絡進行訓練，得到第二識別模型。