[發明專利]目標檢測方法及裝置

在審

• 申請號: 202210719713.8
• 申請日: 2022-06-23
• 公開（公告）號: CN115100392A
• 公開（公告）日: 2022-09-23
• 發明（設計）人: 賀克賽;程新景;楊睿剛
• 申請（專利權）人: 際絡科技（上海）有限公司
• 主分類號: G06V10/25
• 分類號: G06V10/25;G06N3/04;G06N3/08;G06V10/44;G06V10/764;G06V10/774;G06V10/82
• 代理公司: 北京路浩知識產權代理有限公司 11002
• 代理人: 王文君

本發明提供一種目標檢測方法及裝置，方法包括：獲取待檢測圖像；將待檢測圖像輸入至目標檢測模型中，目標檢測模型是基于訓練樣本中目標對于參照區域的位置信息訓練得到；通過目標檢測模型，基于預設掩膜位置得到待檢測圖像中目標相對于待檢測圖像中的參照區域的位置信息；基于位置信息進行分類處理，確定位置分類信息，作為目標檢測結果。本發明通過目標檢測模型，基于預設掩膜位置得到目標相對于參照區域的位置信息，以便于快速對位置信息進行分類處理，得到目標檢測結果，提高了檢測效率；通過訓練樣本中目標相對于參照區域的位置信息對模型進行訓練，以避免目標由于受位置偏移的影響造成檢測不準確的情況，提高檢測精度。

技術領域

本發明涉及計算機視覺技術領域，尤其涉及一種目標檢測方法及裝置。

背景技術

隨著經濟發展，城市建設速度加快，汽車的數量逐年增加。汽車持續快速增長為城市交通系統帶來巨大負擔，違法駕駛和不良駕駛行為依舊存在，駕駛員未佩戴安全帶直接關系到自身和他人的生命安全。安全帶檢測是智能交通領域中車輛違章檢測的一個重要內容，可有效地幫助交通執法人員對駕駛人員佩戴安全帶行為的監督工作。

目前的安全帶檢測方法，基于成本以及日間夜間檢測功能方面的考慮，主要采用紅外攝像頭獲取駕駛艙內的駕駛員圖像，并通過直接將駕駛員圖像中的安全帶作為目標進行檢測。然而，駕駛艙內紅外攝像頭由于受復雜光線以及缺少安全帶色彩等問題，容易造成安全帶檢測困難，以致出現漏檢和誤檢的情形，使得駕駛員檢測系統無法正常工作。

發明內容

本發明提供一種目標檢測方法及裝置，用以解決現有技術中由于光線、色彩等原因造成安全帶檢測困難的缺陷，提高模型檢測的準確率，從而降低誤檢率和漏檢率。

本發明提供一種目標檢測方法，包括：獲取待檢測圖像；將所述待檢測圖像輸入至目標檢測模型中，所述目標檢測模型是基于訓練樣本中目標對于參照區域的位置信息訓練得到；通過所述目標檢測模型，基于預設掩膜位置得到所述待檢測圖像中目標相對于所述待檢測圖像中的參照區域的位置信息；基于所述位置信息進行分類處理，確定位置分類信息，作為目標檢測結果。

根據本發明提供的一種目標檢測方法，基于預設掩膜位置得到所述待檢測圖像中目標相對于所述待檢測圖像中的參照區域的位置信息，包括：對所述待檢測圖像進行特征提取，得到所述待檢測圖像中目標相對于所述預設掩膜位置的平移信息和尺度變換信息；基于所述平移信息、所述尺度變換信息以及所述預設掩膜位置，確定所述待檢測圖像中目標相對于所述待檢測圖像中的參照區域的位置信息。

根據本發明提供的一種目標檢測方法，基于所述位置信息進行分類處理，確定位置分類信息，包括：基于所述位置信息進行分類處理，得到所述參照區域中存在所述目標的第一類別，或者所述參照區域中不存在所述目標的第二類別。

根據本發明提供的一種目標檢測方法，所述目標為安全帶區域，所述參照區域為駕駛員區域。

根據本發明提供的一種目標檢測方法，該方法還包括訓練所述目標檢測模型，包括：獲取訓練樣本及所述訓練樣本對應的目標真值和位置真值；根據所述訓練樣本和所述訓練樣本對應的位置真值，得到平均位置真值；基于所述平均位置真值，確定所述預設掩膜位置；將所述訓練樣本作為訓練使用的輸入數據，將所述訓練樣本對應的目標真值和所述位置真值作為標簽，基于所述預設掩膜位置對待訓練模型進行訓練，得到目標檢測模型。

根據本發明提供的一種目標檢測方法，將各訓練圖像執行數次卷積，獲得目標相對于所述預設掩膜位置的平移信息和尺度變換信息；根據所述平移信息、尺度變換信息以及所述預設掩膜位置，得到所述訓練樣本中目標相對于所述訓練樣本中的參照區域的預測位置信息；基于所述預測位置信息，得到目標預測結果；根據所述目標預測結果、所述目標真值、所述預測位置信息和所述位置真值，構建位置損失函數和分類損失函數，并基于位置損失和分類損失函數收斂，結束訓練。