[發明專利]優化行人檢測模型的訓練數據集的方法、裝置及存儲介質

在審

• 申請號: 202011120107.1
• 申請日: 2020-10-19
• 公開（公告）號: CN112232225A
• 公開（公告）日: 2021-01-15
• 發明（設計）人: 張武強;王寶鋒;支蓉;郭子杰
• 申請（專利權）人: 戴姆勒股份公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/04
• 代理公司: 北京永新同創知識產權代理有限公司 11376
• 代理人: 楊勝軍

1.一種用于優化用于行人檢測模型的訓練數據集的方法，其中，所述方法包括下述步驟：

獲取原始的行人數據，所述原始的行人數據至少包括圖像數據、行人位置數據和邊界框尺寸數據；

獲取原始的行人數據的數據分布特性；

利用原始的行人數據訓練行人檢測模型，從而獲得原始檢測結果；

利用原始的行人數據訓練數據生成模型，所述數據生成模型為生成式模型；

根據數據分布特性和原始檢測結果，利用數據生成模型生成具有特定的邊界框尺寸數據的行人數據作為附加的行人數據；以及

基于原始的行人數據和附加的行人數據，構建用于行人檢測模型的訓練數據集。

2.如權利要求1所述的方法，其中，所述方法還包括下述步驟：

針對邊界框尺寸對原始檢測結果進行評估，獲得檢測準確度相對較低的邊界框尺寸；以及

根據檢測準確度相對較低的邊界框尺寸，確定所述特定的邊界框尺寸數據。

3.如權利要求2所述的方法，其中，檢測準確度相對較低的邊界框尺寸被確定為檢測準確度低于平均檢測準確度的邊界框尺寸。

4.如權利要求1-3中任一項所述的方法，其中，所述方法還包括下述步驟：

針對邊界框尺寸數據分析原始的行人數據的數據分布特性，獲得數據數量相對較少的邊界框尺寸；以及

根據數據數量相對較少的邊界框尺寸，確定所述特定的邊界框尺寸數據。

5.如權利要求4所述的方法，其中，數據數量相對較少的邊界框尺寸被確定為與邊界框尺寸數據的期望值相差大于邊界框尺寸數據的標準差的邊界框尺寸。

6.如權利要求4或5所述的方法，其中，確定所述特定的邊界框尺寸數據的步驟包括：將所述特定的邊界框尺寸數據確定為檢測準確度相對較低并且數據數量相對較少的邊界框尺寸。

7.如權利要求1-6中任一項所述的方法，其中，利用數據生成模型生成具有特定的邊界框尺寸數據的行人數據的步驟包括：使生成的行人數據的行人位置數據符合正態分布。

8.如權利要求7所述的方法，其中，所述方法還包括：利用所述訓練數據集訓練行人檢測模型，從而獲得檢測結果，當所述檢測結果優于原始檢測結果時，確定所述訓練數據集為優化的訓練數據集合，其中：

調整所述正態分布的期望值和/或標準差以獲得不同的訓練數據集，直到獲得優于原始檢測結果的檢測結果；和/或

選擇附加的行人數據的不同的子集用于與原始的行人數據構建不同的訓練數據集，直到獲得優于原始檢測結果的檢測結果。

9.一種用于優化用于行人檢測模型的訓練數據集的優化裝置，所述優化裝置配置成能夠執行根據權利要求1-8中任一項所述的方法，其中，所述優化裝置包括：

行人數據分布分析單元(1)，其配置成能夠獲取原始的行人數據的數據分布特性；

檢測評估單元(2)，其配置成能夠利用原始的行人數據訓練行人檢測模型，從而獲得原始檢測結果；

極端行人姿態挖掘單元(3)，其配置成能夠根據數據分布特性和原始檢測結果，確定特定的邊界框尺寸數據；以及

數據生成單元(4)，其配置成能夠利用原始的行人數據訓練數據生成模型，并利用數據生成模型生成具有特定的邊界框尺寸數據的行人數據，從而構建用于行人檢測模型的訓練數據集。

10.一種計算機可讀存儲介質，其中，所述計算機可讀存儲介質中存儲有計算器程序，當所述計算機程序被一個或多于一個處理器執行時，所述處理器能夠執行根據權利要求1-8中任一項所述的方法。