[發明專利]目標檢測定位置信度確定方法、裝置、電子設備及存儲介質

有效

• 申請號: 202011572377.6
• 申請日: 2020-12-25
• 公開（公告）號: CN112668573B
• 公開（公告）日: 2022-05-10
• 發明（設計）人: 樓文杰;王瑞
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: G06V10/25
• 分類號: G06V10/25;G06V10/40;G06V10/764;G06V10/82;G06K9/62;G06N3/04;G06N3/08;G06T3/40;G06T7/11
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 熊永強

1.一種目標檢測定位置信度確定方法，其特征在于，包括：

獲取待檢測圖像；

將所述待檢測圖像輸入到目標檢測網絡進行目標檢測，得到至少一個候選框、每個候選框的第一掩碼圖以及N個第一特征圖，所述N個第一特征圖為所述目標檢測網絡的第一網絡層的N個通道上的N個特征圖，所述第一網絡層位于所述目標檢測網絡的底部，具體為：所述第一網絡層在所述目標檢測網絡中的位置滿足以下公式：其中，k表示所述第一網絡層在所述目標檢測網絡包含的多個網絡層中的位置，k₀為預設的超參，W₁和H₁分別為所述每個候選框的第二掩碼圖的寬和高，W₀和H₀為所述目標檢測網絡所規定的尺寸，N為大于1的整數；

對所述每個候選框的第一掩碼圖進行擴充，得到所述每個候選框的第二掩碼圖；

從所述N個第一特征圖中截取N個第二特征圖；

將所述每個候選框的第二掩碼圖與所述N個第二特征圖進行拼接，得到所述每個候選框對應的第一目標特征圖，其中，所述第一目標特征圖是一個維度為W₁*H₁*(N+1)的三維矩陣，W₁*H₁為所述每個候選框的第二掩碼圖以及每個第二特征圖的尺寸；

根據所述每個候選框對應的第一目標特征圖，確定所述每個候選框的定位置信度。

2.根據權利要求1所述的方法，其特征在于，在從N個第一特征圖中截取N個第二特征圖之前，所述方法還包括：

獲取所述目標檢測網絡所規定的尺寸；

根據所述每個候選框的第二掩碼圖的尺寸、所述目標檢測網絡所規定的尺寸以及預設的超參，確定所述目標檢測網絡的第一網絡層。

3.根據權利要求1或2所述的方法，其特征在于，所述根據所述每個候選框對應的第一目標特征圖，確定所述每個候選框的定位置信度，包括：

對所述每個候選框對應的第一目標特征圖進行特征提取，得到所述每個候選框對應的第二目標特征圖；

通過兩個卷積核尺寸為1*1的卷積層先后對所述每個候選框對應的第二目標特征圖進行卷積處理，得到所述每個候選框對應的特征向量；

將所述每個候選框對應的特征向量進行歸一化處理，得到所述每個候選框的定位置信度。

4.根據權利要求1所述的方法，其特征在于，在根據所述目標特征圖，確定所述每個候選框的定位置信度之后，所述方法還包括：

獲取所述每個候選框的分類置信度；

根據所述每個候選框的分類置信度以及所述每個候選框的定位置信度，確定所述每個候選框的目標置信度；

根據所述每個候選框的目標置信度，對所述至少一個候選框進行非極大值抑制，得到所述至少一個候選框中的目標候選框。

5.根據權利要求4所述的方法，其特征在于，所述待檢測圖像包括醫學圖像。

6.根據權利要求5所述的方法，其特征在于，所述將所述待檢測圖像輸入到目標檢測網絡進行目標檢測，得到至少一個候選框以及所述至少一個候選框中每個候選框的第一掩碼圖，包括：

將所述醫學圖像輸入到目標檢測網絡，得到所述醫學圖像的第三特征圖；

獲取與所述醫學圖像所屬的醫學領域對應的醫學知識圖譜；

將所述醫學知識圖譜進行向量化，得到與所述醫學知識圖譜對應的第一特征向量；

將所述第一特征向量進行多次復制與組合，得到第一三維矩陣，其中，所述第一三維矩陣中每層的數據相同，且所述第一三維矩陣的每層的尺寸與所述第三特征圖的尺寸相同；

將所述第一三維矩陣與所述醫學圖像的第三特征圖進行拼接，得到第四特征圖，根據所述第四特征圖對所述醫學圖像進行目標檢測，得到所述醫學圖像中與目標對應的至少一個候選框以及所述每個候選框對應的第一掩碼圖。