[發明專利]物體檢測方法、模型訓練方法、裝置及電子設備

有效

• 申請號: 202110672301.9
• 申請日: 2021-06-17
• 公開（公告）號: CN113326796B
• 公開（公告）日: 2022-11-29
• 發明（設計）人: 錢正宇;袁正雄;李金麒;褚振方;黃悅;李潤青;胡鳴人;施恩
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06V20/10
• 分類號: G06V20/10;G06V20/40;G06V10/44;G06V10/764
• 代理公司: 北京銀龍知識產權代理有限公司 11243
• 代理人: 黃燦;劉念

1.一種物體檢測方法，包括：

獲取待檢測圖像；

對所述待檢測圖像進行物體檢測，得到所述待檢測圖像的第一物體檢測信息；

對所述待檢測圖像進行場景檢測，以獲取所述待檢測圖像的目標場景檢測信息，所述目標場景檢測信息包括所述待檢測圖像對應的目標場景類別；

獲取所述目標場景類別對應的場景識別模型，所述場景識別模型用于獲取所述目標場景類別下的場景識別信息，所述場景識別信息用于表征所述目標場景類別對應的場景框中物體的場景環境；

基于所述場景識別模型和所述第一物體檢測信息，確定所述待檢測圖像的第二物體檢測信息，所述第二物體檢測信息用于表征物體在所述目標場景類別的場景下的擺放信息；

所述基于所述場景識別模型和所述第一物體檢測信息，確定所述待檢測圖像的第二物體檢測信息，包括：

基于所述場景識別模型對所述待檢測圖像進行場景識別，得到所述目標場景類別下的場景識別信息；

對所述第一物體檢測信息和所述場景識別信息進行融合處理，得到所述第二物體檢測信息；

所述基于所述場景識別模型對所述待檢測圖像進行場景識別，得到所述目標場景類別下的場景識別信息，包括：

將標記有所述目標場景類別對應的場景框的待檢測圖像輸入至場景識別模型進行場景識別，得到所述目標場景類別下的場景識別信息；

所述對所述待檢測圖像進行場景檢測，以獲取所述待檢測圖像的目標場景檢測信息，包括：

基于場景檢測模型對所述待檢測圖像進行場景檢測，以獲取所述待檢測圖像的目標場景檢測信息；所述場景檢測模型為基于行業場景庫中行業類別下的場景圖像樣本數據訓練得到的模型，所述行業場景庫包括M個行業類別下的場景圖像樣本數據，M為正整數；

所述基于場景檢測模型對所述待檢測圖像進行場景檢測，以獲取所述待檢測圖像的目標場景檢測信息之前，所述方法還包括：

獲取所述待檢測圖像對應的目標行業類別；

從預設場景檢測模型中獲取所述目標行業類別對應的所述場景檢測模型。

2.根據權利要求1所述的方法，其中，所述行業場景庫包括每個行業類別對應的N個場景類別下的場景圖像樣本數據，N為正整數，所述場景識別模型為基于目標數據訓練得到的模型，所述目標數據包括所述行業場景庫中所述目標行業類別的所述目標場景類別下的場景圖像樣本數據，和/或，輸入的所述目標場景類別下的物體圖像樣本數據。

3.一種模型訓練方法，包括：

獲取目標數據，所述目標數據包括目標場景類別下的場景圖像樣本數據，所述目標場景類別為對待檢測圖像進行場景檢測所獲取的目標場景檢測信息中的場景類別；

基于所述目標數據對所述目標場景類別對應的場景識別模型進行訓練；

其中，所述場景識別模型用于結合第一物體檢測信息確定所述待檢測圖像的第二物體檢測信息，所述第一物體檢測信息為對所述待檢測圖像進行物體檢測得到的物體檢測信息，所述場景識別模型用于獲取所述目標場景類別下的場景識別信息，所述場景識別信息用于表征所述目標場景類別對應的場景框中物體的場景環境，所述第二物體檢測信息用于表征物體在所述目標場景類別的場景下的擺放信息；

所述待檢測圖像的第二物體檢測信息通過以下方式確定：

基于所述場景識別模型對所述待檢測圖像進行場景識別，得到所述目標場景類別下的場景識別信息；

對所述第一物體檢測信息和所述場景識別信息進行融合處理，得到所述第二物體檢測信息；

所述基于所述場景識別模型對所述待檢測圖像進行場景識別，得到所述目標場景類別下的場景識別信息，包括：

將標記有所述目標場景類別對應的場景框的待檢測圖像輸入至場景識別模型進行場景識別，得到所述目標場景類別下的場景識別信息；

所述目標場景類別為基于場景檢測模型對待檢測圖像進行場景檢測所獲取的目標場景檢測信息中的場景類別，所述目標場景類別為目標行業類別下的場景類別，所述目標行業類別為所述待檢測圖像對應的行業類別；

所述方法還包括：

從行業場景庫中獲取所述目標行業類別下的場景圖像樣本數據；

基于所述目標行業類別下的場景圖像樣本數據對所述場景檢測模型進行訓練，其中，所述行業場景庫包括M個行業類別下的場景圖像樣本數據，所述M個行業類別包括所述目標行業類別，M為正整數。