[發明專利]候選框處理方法、裝置、設備及介質

在審

• 申請號: 202011381452.0
• 申請日: 2020-11-30
• 公開（公告）號: CN112597994A
• 公開（公告）日: 2021-04-02
• 發明（設計）人: 黎澤明
• 申請（專利權）人: 北京邁格威科技有限公司
• 主分類號: G06K9/32
• 分類號: G06K9/32;G06K9/62
• 代理公司: 北京潤澤恒知識產權代理有限公司 11319
• 代理人: 王婷婷

1.一種候選框處理方法，其特征在于，應用于目標檢測網絡，所述目標檢測網絡包括候選框生成模塊和目標檢測模塊，所述方法包括：

預測所述目標檢測模塊對所述候選框生成模塊輸出的各個候選框的喜好程度；

根據所述目標檢測網絡的當前使用階段，將所述候選框生成模塊輸出的各個候選框中相應喜好程度的候選框輸入到所述目標檢測模塊；其中，在不同的使用階段，輸入到所述目標檢測模塊的候選框的喜好程度不同。

2.根據權利要求1所述的方法，其特征在于，根據所述目標檢測網絡的當前使用階段，將所述候選框生成模塊輸出的各個候選框中相應喜好程度的候選框輸入到所述目標檢測模塊，包括：

在所述目標檢測網絡的當前使用階段為訓練階段的情況下，從所述候選框生成模塊輸出的各個候選框中，采樣喜好程度低于預設閾值的候選框并輸入到所述目標檢測模塊。

3.根據權利要求1所述的方法，其特征在于，根據所述目標檢測網絡的當前使用階段，將所述候選框生成模塊輸出的各個候選框中相應喜好程度的候選框輸入到所述目標檢測模塊，包括：

在所述目標檢測網絡的當前使用階段為應用階段的情況下，從所述候選框生成模塊輸出的各個候選框中，采樣喜好程度不低于所述預設閾值的候選框并輸入到所述目標檢測模塊。

4.根據權利要求1-3任一項所述的方法，其特征在于，所述方法還包括：

在所述候選框生成模塊的輸出端串聯預測模塊；

在所述目標檢測網絡的當前使用階段為訓練階段的情況下，根據所述目標檢測模塊的目標檢測損失值，生成喜好程度標簽；

根據所述預測模塊輸出的喜好程度預測結果與所述喜好程度標簽，對所述預測模塊進行訓練，得到喜好程度預測模塊，所述喜好程度預測模塊用于預測所述目標檢測模塊對所述候選框生成模塊輸出的各個候選框的喜好程度。

5.根據權利要求4所述的方法，其特征在于，所述目標檢測模塊包括目標分類子模塊和目標定位子模塊；根據所述目標檢測模塊的目標檢測損失值，生成喜好程度標簽，包括：

根據所述目標分類子模塊的分類損失值，生成喜好程度標簽；或

根據所述目標分類子模塊的分類損失值和所述目標定位子模塊的定位損失值，生成喜好程度標簽。

6.根據權利要求4所述的方法，其特征在于，根據所述預測模塊輸出的喜好程度預測結果與所述喜好程度標簽，對所述預測模塊進行訓練，得到喜好程度預測模塊，包括：

根據所述預測模塊輸出的喜好程度預測結果與所述喜好程度標簽，確定喜好程度損失值；

根據所述喜好程度損失值，對所述預測模塊進行參數更新。

7.根據權利要求6所述的方法，其特征在于，在所述目標檢測網絡的當前使用階段為訓練階段的情況下，所述方法還包括：

根據所述目標檢測模塊輸出的目標預測結果與所述候選框生成模塊輸出的候選框對應的標簽，生成所述目標檢測損失值；

根據所述喜好程度損失值和所述目標檢測損失值，對所述候選框生成模塊和所述目標檢測模塊進行參數更新。

8.根據權利要求1-7任一所述的方法，其特征在于，所述方法還包括：

在所述目標檢測網絡的當前使用階段為應用階段的情況下，獲得所述目標檢測模塊輸出的對相應的候選框進行檢測的檢測結果；

輸出所述檢測結果。

9.一種候選框處理裝置，其特征在于，應用于目標檢測網絡，所述目標檢測網絡包括候選框生成模塊和目標檢測模塊，所述裝置包括：

預測模塊，用于預測所述目標檢測模塊對所述候選框生成模塊輸出的各個候選框的喜好程度；

輸入模塊，用于根據所述目標檢測網絡的當前使用階段，將所述候選框生成模塊輸出的各個候選框中相應喜好程度的候選框輸入到所述目標檢測模塊；其中，在不同的使用階段，輸入到所述目標檢測模塊的候選框的喜好程度不同。