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

有效

• 申請號: 202110718100.8
• 申請日: 2021-06-28
• 公開（公告）號: CN113378834B
• 公開（公告）日: 2023-08-04
• 發明（設計）人: 何悅;譚嘯;孫昊
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06V20/62
• 分類號: G06V20/62;G06V30/18;G06V30/19;G06V10/764;G06V10/82;G06N3/0464;G06N3/084
• 代理公司: 北京英賽嘉華知識產權代理有限責任公司 11204
• 代理人: 王達佐;馬曉亞

本公開的實施例公開了目標檢測方法、裝置、設備、存儲介質以及程序產品，涉及人工智能領域，具體為計算機視覺和深度學習技術，可應用于智能交通和智慧城市等場景下。該方法的一具體實施方式包括：獲取包括待檢測目標的圖像；將圖像輸入至預先訓練的斜框檢測模型，得到斜框檢測結果，其中，斜框檢測模型包括特征提取網絡和斜框確定網絡，特征提取網絡用于提取圖像的特征，斜框確定網絡用于根據圖像的特征生成像素級的斜框檢測結果，斜框檢測結果用于指示待檢測目標所在的斜框的位置和置信度；根據像素級的斜框檢測結果，確定待檢測目標在圖像中的位置。該實施方式實現了對任意方向斜框的目標檢測。

技術領域

本公開的實施例涉及人工智能領域，具體涉及計算機視覺和深度學習技術，可以應用于智能交通和智慧城市等場景下。

背景技術

目標檢測是計算機視覺和數字圖像處理的一個熱門方向，廣泛應用于各種領域。例如，工業檢測、智能導航、視頻監控、智慧交通等等。目標檢測經常作為一種基礎算法廣泛應用于各種任務中。例如，可以將目標檢測應用于人臉識別、文本識別、實例分割等等。目標檢測算法對后續應用的任務起著至關重要的作用。

由于實際中目標檢測的場景多種多樣，且待檢測目標的標注框常常不是邊均在水平或豎直方向的正框，而是斜框。因此，針對斜框的目標檢測是主要的一個研究方向之一。現有的針對斜框的目標檢測包主要包括基于Anchor-Based的目標檢測方法和基于Anchor-Free的目標檢測方法。

其中，基于Anchor-Based的目標檢測方法一般是在正框檢測模型中增加斜框檢測的分支，通過預先設置的Anchor以生成各種不同角度的旋轉Anchor來實現斜框檢測，或者通過修正正框而得到斜框。基于Anchor-Free的目標檢測方法一般是使用全卷積網絡直接預測目標所在的斜框。

發明內容

本公開的實施例提出了目標檢測方法、裝置、設備、存儲介質以及程序產品。

第一方面，本公開的實施例提供了一種目標檢測方法，該方法包括：獲取包括待檢測目標的圖像；將圖像輸入至預先訓練的斜框檢測模型，得到斜框檢測結果，其中，斜框檢測模型包括特征提取網絡和斜框確定網絡，特征提取網絡用于提取圖像的特征，斜框確定網絡用于根據圖像的特征生成像素級的斜框檢測結果，斜框檢測結果用于指示待檢測目標所在的斜框的位置和置信度；根據像素級的斜框檢測結果，確定待檢測目標在圖像中的位置。

第二方面，本公開的實施例提供了一種目標檢測裝置，該裝置包括：圖像獲取模塊，被配置成獲取包括待檢測目標的圖像；斜框檢測模塊，被配置成將圖像輸入至預先訓練的斜框檢測模型，得到斜框檢測結果，其中，斜框檢測模型包括特征提取網絡和斜框確定網絡，特征提取網絡用于提取圖像的特征，斜框確定網絡用于根據圖像的特征生成像素級的斜框檢測結果，斜框檢測結果用于指示待檢測目標所在的斜框的位置和置信度；位置確定模塊，被配置成根據像素級的斜框檢測結果，確定待檢測目標在圖像中的位置。

第三方面，本公開實施例提出了一種電子設備，包括：至少一個處理器；以及與至少一個處理器通信連接的存儲器；其中，存儲器存儲有可被至少一個處理器執行的指令，指令被至少一個處理器執行，以使至少一個處理器能夠執行如第一方面中任一實現方式描述的方法。

第四方面，本公開實施例提出了一種存儲有計算機指令的非瞬時計算機可讀存儲介質，計算機指令用于使計算機執行如第一方面中任一實現方式描述的方法。

第五方面，本公開實施例提出了一種計算機程序產品，包括計算機程序，計算機程序在被處理器執行時實現如第一方面中任一實現方式描述的方法。