[發明專利]一種標簽分配方法及裝置

在審

• 申請號: 202210008701.4
• 申請日: 2022-01-06
• 公開（公告）號: CN114495053A
• 公開（公告）日: 2022-05-13
• 發明（設計）人: 趙楊楊;黃暢
• 申請（專利權）人: 北京地平線信息技術有限公司
• 主分類號: G06V20/58
• 分類號: G06V20/58;G06N3/04;G06N3/08
• 代理公司: 北京弘權知識產權代理有限公司 11363
• 代理人: 郭放;許偉群

本公開實施例公開了一種標簽分配方法及裝置，其中，標簽分配方法包括：計算圖像中的樣本點針對至少一個標記框的置信度；基于所述樣本點與所述標記框的交并比，從所述樣本點中為所述標記框選擇k個候選點，其中，k為預設的正整數；基于所述樣本點針對所述標記框的置信度，為所述k個候選點分配標簽。本公開實施例在進行標簽分配時，結合了樣本點針對標記框的置信度和與標記框交并比，能夠提高標簽分配的準確度。

技術領域

本公開涉及目標檢測技術領域，尤其是一種標簽分配方法及裝置。

背景技術

標簽分配是目標檢測在模型訓練階段的一個重要步驟，通常用于區分模型訓練階段中的各個樣本的正負屬性。

但是，目前的標簽分配技術往往準確度較低，進而會影響到目標檢測的準確度。因此，如何提供一種準確度較高的標簽分配方法成為亟需解決的問題。

發明內容

為了解決上述技術問題，提出了本公開。本公開的實施例提供了一種標簽分配方法及裝置。

根據本公開實施例的一個方面，提供了一種標簽分配方法，包括：

計算圖像中的樣本點針對至少一個標記框的置信度；

基于所述樣本點與所述標記框的交并比，從所述樣本點中為所述標記框選擇k個候選點，其中，k為預設的正整數；

基于所述樣本點針對所述標記框的置信度，為所述k個候選點分配標簽。

根據本公開實施例的又一個方面，提供了一種標簽分配裝置，包括：

置信度計算模塊，用于計算圖像中的樣本點針對至少一個標記框的置信度；

候選點確定模塊，用于基于所述樣本點與所述標記框的交并比，從所述樣本點中為所述標記框選擇k個候選點，k為預設的正整數；

標簽分配模塊，用于基于所述置信度計算模塊得到的所述樣本點針對所述標記框的置信度，為所述候選點確定模塊確定的所述k個候選點分配標簽。

根據本公開實施例的又一個方面，提供了一種計算機可讀存儲介質，所述存儲介質存儲有計算機程序，所述計算機程序用于執行本公開上述任一實施例所述的標簽分配方法。

根據本公開實施例的又一個方面，提供了一種電子設備，所述電子設備包括：

處理器；

用于存儲所述處理器可執行指令的存儲器；

所述處理器，用于從所述存儲器中讀取所述可執行指令，并執行所述指令以實現本公開上述任一實施例所述的標簽分配方法。

基于本公開上述實施例提供的標簽分配方法和裝置、計算機可讀存儲介質和電子設備，在進行標簽分配時，首先計算樣本點針對標記框的置信度，并且基于樣本點與標記框的交并比，從樣本點中為標記框選擇k個候選點，然后基于樣本點針對標記框的置信度，為該k個候選點分配標簽。也就是說，通過本公開的方案進行標簽分配時，結合了樣本點針對標記框的置信度和樣本點與標記框的交并比。

現有技術在進行標簽分配時，只應用樣本點針對標記框的置信度。與現有技術相比，本公開實施例的標簽分配方法在進行標簽分配時，應用的信息更多樣化，因此，本公開實施例的方案能夠提高標簽分配的準確度，進一步的，也能夠提高目標檢測的準確度。

附圖說明

通過結合附圖對本申請實施例進行更詳細的描述，本申請的上述以及其他目的、特征和優勢將變得更加明顯。附圖用來提供對本申請實施例的進一步理解，并且構成說明書的一部分，與本申請實施例一起用于解釋本申請，并不構成對本申請的限制。在附圖中，相同的參考標號通常代表相同部件或步驟。

圖1是本公開所適用的設備示意圖。