[發(fā)明專利]寵物識別方法、裝置、設備及計算機可讀存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202110043107.4
• 申請日: 2021-01-13
• 公開（公告）號: CN112699842A
• 公開（公告）日: 2021-04-23
• 發(fā)明（設計）人: 皮人偉;李靜濤;劉超;馬紹秋;麥劉偉;唐嘉良;江寧;張勝利;袁華;楊柳
• 申請（專利權）人: 上海鏡河科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京輕創(chuàng)知識產(chǎn)權代理有限公司 11212
• 代理人: 厲洋洋

1.一種寵物識別方法，其特征在于，所述寵物識別方法包括：

根據(jù)生成對抗網(wǎng)絡對寵物標記圖像進行增強，得到增強訓練圖像；

對所述增強訓練圖像進行訓練得到輕量級目標檢測模型YOLO和基于加法網(wǎng)絡AdderNet的圖像分類模型；

根據(jù)所述YOLO模型對采集圖像中的寵物進行檢測，得到僅包含寵物的目標圖像；

將所述目標圖像輸入到所述基于AdderNet的圖像分類模型中，得到所述目標圖像中的寵物所屬類別。

2.根據(jù)權利要求1所述的寵物識別方法，其特征在于，所述根據(jù)生成對抗網(wǎng)絡對寵物標記圖像進行增強，得到增強訓練圖像包括：

隨機生成一組噪音向量輸入到生成對抗網(wǎng)絡的生成器中，得到初始生成圖像；

將所述初始生成圖像和寵物標記圖像輸入到生成對抗網(wǎng)絡的判別器中判別真假，以訓練所述生成對抗網(wǎng)絡；

訓練完成后，將所述生成對抗網(wǎng)絡的生成器生成的目標生成圖像和寵物標記圖像作為所述增強訓練圖像。

3.根據(jù)權利要求2所述的寵物識別方法，其特征在于，所述方法還包括：

對所述寵物標記圖像進行數(shù)據(jù)增強處理，得到增強處理圖像；所述數(shù)據(jù)增強處理包括：翻轉(zhuǎn)、旋轉(zhuǎn)、灰度化和裁剪；

將所述增強處理圖像、目標生成圖像和寵物標記圖像混合得到所述增強訓練圖像。

4.根據(jù)權利要求1所述的寵物識別方法，其特征在于，所述根據(jù)所述YOLO模型對采集圖像中的寵物進行檢測之前包括：

將當前攝像頭拍攝獲取的圖像作為所述采集圖像；或，從服務器獲取的圖像作為所述采集圖像。

5.根據(jù)權利要求1所述的寵物識別方法，其特征在于，所述根據(jù)所述YOLO模型對所述采集圖像進行檢測，得到僅包含寵物的目標圖像包括：

將所述采集圖像輸入到YOLO模型，輸出所述采集圖像中寵物所在坐標值和置信度；

當所述置信度大于預設預設置信度時，根據(jù)與所述置信度對應的寵物所在坐標值分離所述采集圖像得到所述目標圖像。

6.根據(jù)權利要求1所述的寵物識別方法，其特征在于，所述得到所述目標圖像中的寵物所屬類別，包括：

對所述增強訓練圖像進行訓練得到圖像識別模型MobileNet；

根據(jù)所述MobileNet對所述目標圖像中的寵物所屬的類別進行識別，得到第一識別結果；

將所述目標圖像輸入到所述基于AdderNet的圖像分類模型中，得到第二識別結果；

當所述第一識別結果和所述第二識別結果相同時，確定所述目標圖像中的寵物所屬類別。

7.根據(jù)權利要求1-6任一項所述的寵物識別方法，其特征在于，所述確定所述目標圖像中的寵物所屬類別之后，包括：

當確定所述目標圖像中的寵物獨自時，根據(jù)所述寵物所屬類別和所述目標圖像查找所述寵物的聯(lián)系人信息；

通過所述寵物的聯(lián)系人信息聯(lián)系所述對應的聯(lián)系人。

8.一種寵物識別裝置，其特征在于，所述寵物識別裝置包括圖像增強模塊、訓練模塊、圖像檢測模塊和圖像識別模塊；

所述圖像增強模塊，用于根據(jù)生成對抗網(wǎng)絡對原始寵物標記圖像進行增強，得到增強訓練圖像；

所述訓練模型，用于對所述增強訓練圖像進行訓練得到神經(jīng)網(wǎng)絡模型YOLO和基于加法網(wǎng)絡AdderNet的圖像分類模型；

所述圖像檢測模塊，用于根據(jù)所述YOLO模型對采集圖像中的寵物進行檢測，得到僅包含寵物的目標圖像；

所述圖像識別模塊，用于將所述目標圖像輸入到所述基于AdderNet的圖像分類模型中，得到所述目標圖像中的寵物所屬類別。

9.一種寵物識別設備，其特征在于，所述寵物識別設備包括存儲器和處理器；

所述處理器用于對執(zhí)行存儲器中存儲的一個或者多個計算機程序，以實現(xiàn)根據(jù)權利要求1至7中任一項所述的寵物識別方法的步驟。

10.一種計算機可讀存儲介質(zhì)，其特征在于，所述計算機可讀存儲介質(zhì)存儲有一個或者多個計算機程序，所述一個或者多個計算機程序可被一個或者多個處理器執(zhí)行，以實現(xiàn)根據(jù)權利要求1至7中任一項所述的寵物識別方法的步驟。