[發明專利]圖像檢測方法、裝置、電子設備及可讀存儲介質

在審

• 申請號: 202010613093.0
• 申請日: 2020-06-30
• 公開（公告）號: CN111783639A
• 公開（公告）日: 2020-10-16
• 發明（設計）人: 戴兵;葉芷;李揚曦
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/46;G06N3/04;G06N3/08
• 代理公司: 北京同立鈞成知識產權代理有限公司 11205
• 代理人: 張娜;劉芳

1.一種圖像檢測方法，包括：

獲取待檢測圖像；

根據所述待檢測圖像，基于預先訓練的標志檢測網絡，獲得所述待檢測圖像中的至少一個候選區域，所述候選區域包含預設標志；

根據所述至少一個候選區域，基于預先訓練的標志識別網絡，獲得所述至少一個候選區域的特征；

根據所述至少一個候選區域的特征，確定所述待檢測圖像的類別。

2.根據權利要求1所述的方法，其中，所述標志檢測網絡包括預先訓練的特征提取網絡和預先訓練的候選區域產生網絡；

所述根據所述待檢測圖像，基于預先訓練的標志檢測網絡，獲得所述待檢測圖像中的至少一個候選區域，包括：

將所述待檢測圖像輸入所述特征提取網絡，提取所述待檢測圖像的關鍵特征；

將所述待檢測圖像的關鍵特征輸入所述候選區域產生網絡，獲得所述至少一個候選區域的位置坐標。

3.根據權利要求2所述的方法，其中，所述候選區域產生網絡包括回歸模型和分類模型；

所述將所述待檢測圖像的關鍵特征輸入所述候選區域產生網絡，獲得所述至少一個候選區域的位置坐標，包括：

將所述待檢測圖像的關鍵特征輸入所述回歸模型，獲得至少一個待處理區域的位置坐標；

針對每個所述待處理區域，將所述待處理區域的關鍵特征輸入所述分類模型，確定所述待處理區域是否包含所述預設標志；當確定所述待處理區域包含所述預設標志時，將所述待處理區域的位置坐標作為所述候選區域的位置坐標。

4.根據權利要求1所述的方法，其中，所述根據所述至少一個候選區域，基于預先訓練的標志識別網絡，獲得所述至少一個候選區域的特征，包括：

根據所述至少一個候選區域，基于所述待檢測圖像獲得與所述至少一個候選區域分別對應的至少一張區域圖像；

針對每張區域圖像，將所述區域圖像輸入所述標志識別網絡，確定所述區域圖像的特征作為與所述區域圖像對應的候選區域的特征。

5.根據權利要求4所述的方法，其中，所述根據所述至少一個候選區域的特征，確定所述待檢測圖像的類別，包括：

針對每張區域圖像，將所述區域圖像的特征與預設備選特征集中的各備選特征進行比對，確定所述區域圖像包含的所述預設標志的類別；

根據所述至少一張區域圖像包含的所述預設標志的類別，確定所述待檢測圖像的類別。

6.根據權利要求5所述的方法，其中，所述將所述區域圖像的特征與預設備選特征集中的各備選特征進行比對之前，還包括：

將預設備選標志集中的各備選標志輸入所述標志識別網絡，獲得各所述備選標志的備選特征；

用各所述備選標志的備選特征構建所述預設備選特征集。

7.根據權利要求6所述的方法，其中，所述將所述區域圖像的特征與預設備選特征集中的各備選特征進行比對，確定所述區域圖像包含的所述預設標志的類別，包括：

分別確定所述區域圖像的特征與各所述備選特征之間的歐氏距離；

當備選特征與所述區域圖像的特征之間的歐氏距離小于預設閾值時，將與所述備選特征所對應的備選標志的類別確定為所述區域圖像包含的所述預設標志的類別。

8.根據權利要求1-7中任一項所述的方法，其中，所述獲取待檢測圖像，包括：

獲取待檢測視頻；

抽取所述待檢測視頻的關鍵幀圖像，并將所述關鍵幀圖像作為所述待檢測圖像。

9.根據權利要求1-7中任一項所述的方法，其中，所述標志檢測網絡為卷積神經網絡。

10.根據權利要求2或3所述的方法，其中，所述特征提取網絡為殘差網絡。

11.根據權利要求1-7中任一項所述的方法，其中，所述標志識別網絡為卷積神經網絡。