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

在審

• 申請號: 202110448199.4
• 申請日: 2021-04-25
• 公開（公告）號: CN113139471A
• 公開（公告）日: 2021-07-20
• 發明（設計）人: 吳凌云;胡志強
• 申請（專利權）人: 上海商湯智能科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京林達劉知識產權代理事務所(普通合伙) 11277
• 代理人: 劉新宇

本公開涉及一種目標檢測方法及裝置、電子設備和存儲介質，其中，所述方法包括：獲取目標圖像；利用目標檢測網絡對所述目標圖像進行檢測，得到所述目標圖像中目標對象的第一檢測結果；其中，所述目標檢測網絡是基于樣本圖像以及樣本圖像的監督信息訓練得到的，以及所述樣本圖像的監督信息包括掩膜圖。本公開實施例可以應用于醫療場景中，可對醫學影像進行目標檢測，提高目標檢測網絡的檢測精度。

技術領域

本公開涉及計算機視覺技術領域，尤其涉及一種目標檢測方法及裝置、電子設備和存儲介質。

背景技術

計算機視覺技術是利用計算機以及相關設備對生物視覺進行模擬的一種技術，通過對采集的圖像或視頻進行處理可以獲得相應場景的信息，從而模擬生物的觀測。目標檢測是計算視覺技術的一個重要任務。

目標檢測可以應用在多種場景，例如，在醫學場景中，可以利用目標檢測對醫學影像的病灶區域進行識別，對醫學影像輔助分析。但是目前目標檢測任務中，檢測結果的精度較低。

發明內容

本公開提出了一種目標檢測技術方案。

根據本公開的一方面，提供了一種目標檢測方法，包括：

獲取目標圖像；利用目標檢測網絡對所述目標圖像進行檢測，得到所述目標圖像中目標對象的第一檢測結果；其中，所述目標檢測網絡是基于樣本圖像以及樣本圖像的監督信息訓練得到的，以及所述樣本圖像的監督信息包括掩膜圖。

在本公開實施例中，可以獲取待檢測的目標圖像，利用目標檢測網絡對目標圖像進行檢測，得到目標圖像中目標對象的第一檢測結果。這里，目標檢測網絡是基于樣本圖像以及樣本圖像的監督信息得到的，樣本圖像的監督信息包括掩膜圖，掩膜圖可以對樣本圖像對應的空間信息進行監督，從而由樣本圖像以及樣本圖像的監督信息得到的目標檢測網絡可以保留目標圖像的空間信息，提高目標檢測網絡的檢測精度。

在一個或多個可能的實現方式中，所述利用目標檢測網絡對所述目標圖像進行檢測，得到所述目標圖像中目標對象的第一檢測結果，包括：基于所述目標檢測網絡，確定至少一個第一候選區域；基于所述至少一個第一候選區域的圖像特征，得到包括所述目標對象的類別信息和位置信息的第一檢測結果。

通過目標檢測網絡可以確定目標對象可能位于的至少一個第一候選區域，可以針對第一候選區域中可能存在的目標對象進行檢測，從而使檢測到的目標對象更加全面。

在一個或多個可能的實現方式中，所述方法還包括：通過一神經網絡對所述樣本圖像進行處理，得到所述樣本圖像中參照對象的第二檢測結果和熱力圖；基于所述第二檢測結果、所述熱力圖以及所述樣本圖像的監督信息，確定所述神經網絡的網絡損失，其中，所述熱力圖用于指示所述參照對象在所述樣本圖像中的預測位置；基于所述網絡損失對所述神經網絡的網絡參數進行調整，得到訓練好的神經網絡，其中，所述訓練好的神經網絡包括所述目標檢測網絡。

這里，由于神經網絡在訓練過程中對提供空間信息的熱力圖進行監督，從而由訓練好的神經網絡得到的目標檢測網絡可具有較高的檢測精度。

在一個或多個可能的實現方式中，所述神經網絡包括檢測分支和分割分支；所述通過一神經網絡對所述樣本圖像進行處理，得到所述樣本圖像中所述參照對象的第二檢測結果和熱力圖，包括：利用所述神經網絡對所述樣本圖像進行特征提取，得到至少一個第二候選區域；將所述至少一個第二候選區域輸入所述檢測分支，得到所述樣本圖像中參照對象的第二檢測結果；將所述至少一個第二候選區域輸入所述分割分支，得到所述參照對象的熱力圖。

這里，由于神經網絡的檢測分支和分割分支共享樣本圖像的圖像特征，分割分支可以保留特征空間的結構信息，從而可以提高檢測分支的檢測精度。分割分支采用放射狀的掩膜圖進行監督，相比于精細標注對象輪廓的完整掩膜圖而言，可以大大減少人工標注成本。