[發明專利]目標檢測方法、裝置、設備以及存儲介質

有效

• 申請號: 202011215947.6
• 申請日: 2020-11-04
• 公開（公告）號: CN112329616B
• 公開（公告）日: 2023-08-11
• 發明（設計）人: 楊喜鵬;譚嘯;孫昊
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06V10/82
• 分類號: G06V10/82;G06V20/40;G06V10/80;G06N3/0464;G06N3/08
• 代理公司: 北京市漢坤律師事務所 11602
• 代理人: 姜浩然;吳麗麗

本公開提供一種目標檢測方法、裝置、設備以及存儲介質，涉及計算機視覺、圖像處理、深度學習等人工智能領域。目標檢測方法包括：獲取待檢測圖像；獲取與待檢測圖像相應的背景圖像；將待檢測圖像和背景圖像分別輸入第一目標檢測子模型，獲取第一目標檢測子模型輸出的待檢測圖像的特征圖和背景圖像的特征圖；以及將待檢測圖像的特征圖和背景圖像的特征圖輸入第二目標檢測子模型，獲取第二目標檢測子模型輸出的目標檢測結果。

技術領域

本公開涉及計算機視覺、圖像處理、深度學習等人工智能領域，更具體地，涉及一種目標檢測方法、裝置、設備以及存儲介質。

背景技術

目標檢測是計算機視覺和圖像處理的一個熱門方向，廣泛應用于自動駕駛、智能視頻監控、工業檢測等諸多領域。通過實現計算機視覺減少對人力資本的消耗，具有重要的現實意義。因此，目標檢測也就成為了近年來理論和應用的研究熱點。由于深度學習的廣泛運用，目標檢測方法得到了快速的發展，但現有的目標檢測方法對場景的感知能力弱，仍有一定的局限性。

在此部分中描述的方法不一定是之前已經設想到或采用的方法。除非另有指明，否則不應假定此部分中描述的任何方法僅因其包括在此部分中就被認為是現有技術。類似地，除非另有指明，否則此部分中提及的問題不應認為在任何現有技術中已被公認。

發明內容

根據本公開的一方面，提供了一種目標檢測方法，包括：獲取待檢測圖像；獲取與待檢測圖像相應的背景圖像；將待檢測圖像和背景圖像分別輸入第一目標檢測子模型，獲取第一目標檢測子模型輸出的待檢測圖像的特征圖和背景圖像的特征圖；以及將待檢測圖像的特征圖和背景圖像的特征圖輸入第二目標檢測子模型，獲取第二目標檢測子模型輸出的目標檢測結果。

根據本公開的另一方面，還提供了一種神經網絡的訓練方法，包括：獲取樣本圖像，并標記樣本圖像包括的目標的真實類別和真實目標框；獲取與樣本圖像相應的背景圖像；將樣本圖像和背景圖像分別輸入第一目標檢測子模型，獲取第一目標檢測子模型輸出的樣本圖像的特征圖和背景圖像的特征圖；將樣本圖像的特征圖和背景圖像的特征圖輸入第二目標檢測子模型，獲取第二目標檢測子模型輸出的預測類別和預測目標框及其預測置信度；基于真實類別、真實目標框、預測類別和預測目標框及其預測置信度，計算損失函數；以及基于損失函數，調整第一目標檢測子模型和第二目標檢測子模型的參數。

根據本公開的另一方面，還提供了一種目標檢測裝置，包括：第一圖像獲取模塊，被配置用于獲取待檢測圖像；第二圖像獲取模塊，被配置用于獲取與待檢測圖像相應的背景圖像；第一目標檢測子模塊，被配置用于分別基于待檢測圖像和背景圖像，輸出待檢測圖像的特征圖和背景圖像的特征圖；以及第二目標檢測子模塊，被配置用于基于待檢測圖像的特征圖和背景圖像的特征圖，輸出目標檢測結果。

根據本公開的另一方面，還提供了一種神經網絡，包括：第一目標檢測子模型，被配置用于分別基于待檢測圖像和背景圖像，輸出待檢測圖像的特征圖和背景圖像的特征圖；以及第二目標檢測子模型，被配置用于基于待檢測圖像的特征圖和背景圖像的特征圖生成的特征圖，輸出目標檢測結果。

根據本公開的另一方面，還提供了一種電子設備，包括：處理器；以及存儲程序的存儲器，上述程序包括指令，并且指令在由處理器執行時使處理器執行根據上述的目標檢測方法和/或根據上述的訓練方法。

根據本公開的另一方面，還提供了一種存儲程序的計算機可讀存儲介質，上述程序包括指令，并且指令在由電子設備的處理器執行時，致使電子設備執行根據上述的目標檢測方法和/或根據上述的訓練方法。

根據本公開的另一方面，還提供了一種計算機程序產品，包括計算機程序，其中，所述計算機程序在被處理器執行時實現上述的目標檢測方法和/或根據上述的訓練方法。

本公開的技術方案通過將待檢測圖像和相應的背景圖像輸入目標檢測模型，讓目標檢測模型學習到更多的場景信息，從而提升目標檢測模型對當前場景的感知能力，進而能夠更好地將待檢測圖片中的待檢測目標與背景圖像進行區分，提高目標檢測模型的準確率。