[發(fā)明專利]目標檢測方法、裝置、電子設(shè)備及存儲介質(zhì)

有效

• 申請?zhí)枺?/td>: 202011574454.1
• 申請日: 2020-12-25
• 公開（公告）號: CN112598070B
• 公開（公告）日: 2023-07-28
• 發(fā)明（設(shè)計）人: 張發(fā)恩;宋亮
• 申請（專利權(quán)）人: 創(chuàng)新奇智（廣州）科技有限公司
• 主分類號: G06V10/75
• 分類號: G06V10/75;G06V10/762;G06V10/774
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 鐘揚飛

1.一種目標檢測方法，其特征在于，包括：

獲取待檢測圖像及目標對象圖像；

輸入所述待檢測圖像至預設(shè)的骨干網(wǎng)絡模型，得到對應的特征圖，并輸入所述目標對象圖像至所述預設(shè)的骨干網(wǎng)絡模型，得到對應的特征向量；

逐點比較所述特征圖與所述特征向量的相似性，得到對應的熱力圖；

根據(jù)所述熱力圖，聚類得到目標對象的輪廓；

根據(jù)所述目標對象的輪廓，檢測得到所述待檢測圖像中目標對象的檢測框；

所述逐點比較所述特征圖與所述特征向量的相似性，得到對應的熱力圖，包括：

基于特征之間的歐式距離或余弦距離，逐點比較所述特征圖與所述特征向量的相似性，得到對應的熱力圖；

所述根據(jù)所述目標對象的輪廓，檢測得到所述待檢測圖像中目標對象的檢測框，包括：

根據(jù)所述目標對象的輪廓，計算得到目標對象的坐標及目標對象的大小；

根據(jù)所述目標對象的坐標及所述目標對象的大小，檢測得到所述待檢測圖像中目標對象的檢測框。

2.根據(jù)權(quán)利要求1所述的目標檢測方法，其特征在于，在所述逐點比較所述特征圖與所述特征向量的相似性，得到對應的熱力圖之后，所述方法還包括：

獲取目標對象的實際熱力圖；

根據(jù)所述熱力圖及所述目標對象的實際熱力圖，計算得到對應的交叉熵，并以所述交叉熵作為所述預設(shè)的骨干網(wǎng)絡模型的損失函數(shù)；

基于所述損失函數(shù)，更新所述預設(shè)的骨干網(wǎng)絡模型的參數(shù)。

3.根據(jù)權(quán)利要求2所述的目標檢測方法，其特征在于，所述基于所述損失函數(shù)，更新所述預設(shè)的骨干網(wǎng)絡模型的參數(shù)，包括：

通過反向傳播，計算得到所述損失函數(shù)的梯度；

以所述損失函數(shù)的梯度，更新所述預設(shè)的骨干網(wǎng)絡模型的參數(shù)。

4.一種目標檢測裝置，其特征在于，包括：

獲取模塊，用于獲取待檢測圖像及目標對象圖像；

特征提取模塊，用于輸入所述待檢測圖像至預設(shè)的骨干網(wǎng)絡模型，得到對應的特征圖，并輸入所述目標對象圖像至所述預設(shè)的骨干網(wǎng)絡模型，得到對應的特征向量；

處理模塊，用于逐點比較所述特征圖與所述特征向量的相似性，得到對應的熱力圖；

聚類模塊，用于根據(jù)所述熱力圖，聚類得到目標對象的輪廓；

檢測模塊，用于根據(jù)所述目標對象的輪廓，檢測得到所述待檢測圖像中目標對象的檢測框；

所述檢測模塊，具體用于：

根據(jù)所述目標對象的輪廓，計算得到目標對象的坐標及目標對象的大小；

根據(jù)所述目標對象的坐標及所述目標對象的大小，檢測得到所述待檢測圖像中目標對象的檢測框；

所述檢測模塊，還具體用于：

根據(jù)目標對象的輪廓，計算得到目標對象的坐標及目標對象的大小；

根據(jù)目標對象的坐標及目標對象的大小，檢測得到待檢測圖像中目標對象的檢測框。

5.根據(jù)權(quán)利要求4所述的目標檢測裝置，其特征在于，所述獲取模塊，還用于獲取目標對象的實際熱力圖；

所述目標檢測裝置還包括：

計算模塊，用于根據(jù)所述熱力圖及所述目標對象的實際熱力圖，計算得到對應的交叉熵，并以所述交叉熵作為所述預設(shè)的骨干網(wǎng)絡模型的損失函數(shù)；

參數(shù)更新模塊，用于基于所述損失函數(shù)，更新所述預設(shè)的骨干網(wǎng)絡模型的參數(shù)。

6.一種電子設(shè)備，其特征在于，包括存儲器及處理器，所述存儲器用于存儲計算機程序，所述處理器運行所述計算機程序以使所述電子設(shè)備執(zhí)行根據(jù)權(quán)利要求1至3中任一項所述的目標檢測方法。

7.一種計算機可讀存儲介質(zhì)，其特征在于，其存儲有計算機程序，所述計算機程序被處理器執(zhí)行時實現(xiàn)如權(quán)利要求1至3中任一項所述的目標檢測方法。