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

在審

• 申請?zhí)枺?/td>: 202210064885.6
• 申請日: 2022-01-20
• 公開（公告）號: CN114399696A
• 公開（公告）日: 2022-04-26
• 發(fā)明（設計）人: 宋沉蔓
• 申請（專利權）人: 中國農(nóng)業(yè)銀行股份有限公司
• 主分類號: G06V20/20
• 分類號: G06V20/20;G06V10/80;G06N3/04;G06N3/08
• 代理公司: 北京品源專利代理有限公司 11332
• 代理人: 趙迎迎

本發(fā)明實施例公開了一種目標檢測方法、裝置、存儲介質(zhì)及電子設備，其中方法包括：獲取待檢測圖像；將待檢測圖像輸入到預先訓練的目標檢測模型中，得到目標檢測模型輸出的目標檢測圖像，其中，目標檢測模型包括特征提取模塊、空間注意模塊和通道注意模塊，所述提取模塊中用于提取不同層次的特征信息，空間注意模塊用于對所述特征提取模塊提取的第一特征信息進行空間特征加權，通道注意模塊用于對所述特征提取模塊提取的第二特征信息進行通道加權，經(jīng)過空間特征加權后的第一特征信息和通道加權后的第二特征信息通過圖像融合得到目標檢測圖像。通過注意力機制關注空間特征和通道特征中的重要特征，并抑制不必要特征，有利于提高目標檢測的準確度。

技術領域

本發(fā)明實施例涉及圖像處理技術領域，尤其涉及一種目標檢測方法、裝置、存儲介質(zhì)及電子設備。

背景技術

顯著性目標檢測作為其他視覺任務的預處理步驟一直是計算機視覺領域的研究熱點。可廣泛應用于圖像自動裁剪、圖像識別和檢測、視頻摘要、目標跟蹤等任務中。

傳統(tǒng)的基于手工特征的顯著性檢測方法主要依賴手工制作的特征，而這些特征可能無法描述復雜的圖像場景和結(jié)構，無法適應新的場景和對象，泛化能力差。

發(fā)明內(nèi)容

本發(fā)明實施例提供一種目標檢測方法、裝置、存儲介質(zhì)及電子設備，以實現(xiàn)提高目標檢測的檢測精度。

第一方面，本發(fā)明實施例提供了一種目標檢測方法，包括：

獲取待檢測圖像；

將所述待檢測圖像輸入到預先訓練的目標檢測模型中，得到所述目標檢測模型輸出的目標檢測圖像，其中，所述目標檢測模型包括特征提取模塊、空間注意模塊和通道注意模塊，所述特征提取模塊中用于提取不同層次的特征信息，所述空間注意模塊用于對所述特征提取模塊提取的第一特征信息進行空間特征加權，所述通道注意模塊用于對所述特征提取模塊提取的第二特征信息進行通道加權，經(jīng)過空間特征加權后的第一特征信息和通道加權后的第二特征信息通過圖像融合得到目標檢測圖像。

第二方面，本發(fā)明實施例還提供了一種目標檢測裝置，包括：

圖像獲取模塊，用于獲取待檢測圖像；

目標檢測模塊，用于將所述待檢測圖像輸入到預先訓練的目標檢測模型中，得到所述目標檢測模型輸出的目標檢測圖像，其中，所述目標檢測模型包括特征提取模塊、空間注意模塊和通道注意模塊，所述特征提取模塊中用于提取不同層次的特征信息，所述空間注意模塊用于對所述特征提取模塊提取的第一特征信息進行空間特征加權，所述通道注意模塊用于對所述特征提取模塊提取的第二特征信息進行通道加權，經(jīng)過空間特征加權后的第一特征信息和通道加權后的第二特征信息通過圖像融合得到目標檢測圖像。

第三方面，本公開實施例還提供了一種電子設備，所述電子設備包括：

一個或多個處理器；

存儲裝置，用于存儲一個或多個程序，

當所述一個或多個程序被所述一個或多個處理器執(zhí)行，使得所述一個或多個處理器實現(xiàn)如本公開實施例任一所述的目標檢測方法。

第四方面，本公開實施例還提供了一種包含計算機可執(zhí)行指令的存儲介質(zhì)，所述計算機可執(zhí)行指令在由計算機處理器執(zhí)行時用于執(zhí)行如本公開實施例任一所述的目標檢測方法。

本實施例的技術方案，通過預先設置包括空間注意模塊和通道注意模塊的目標檢測模型，通過目標檢測模型對待檢測圖像進行目標檢測，使得在對待檢測圖像進行目標檢測的過程中，通過注意力機制關注空間特征和通道特征中的重要特征，并抑制不必要特征，有利于提高目標檢測的準確度。

附圖說明

圖1為本發(fā)明實施例提供的一種目標檢測方法的流程示意圖；

圖2是本發(fā)明實施例提供的一種目標檢測模型的結(jié)構示意圖；