[發明專利]訓練圖像的生成方法及裝置、電子設備和存儲介質

在審

• 申請號: 201911128049.4
• 申請日: 2019-11-18
• 公開（公告）號: CN110909654A
• 公開（公告）日: 2020-03-24
• 發明（設計）人: 王露;朱烽;趙瑞
• 申請（專利權）人: 深圳市商湯科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/32;G06K9/62;G06N3/08
• 代理公司: 北京林達劉知識產權代理事務所(普通合伙) 11277
• 代理人: 劉新宇

1.一種訓練圖像的生成方法，其特征在于，包括：

獲取原始訓練圖像；

在添加對象集合中，隨機選定添加對象子集合；

根據所述添加對象子集合，得到目標添加對象；

根據所述目標添加對象，對所述原始訓練圖像進行圖像處理，得到所述訓練圖像。

2.根據權利要求1所述的方法，其特征在于，所述在添加對象集合中，隨機選定添加對象子集合，包括：

隨機生成第一參考數據；

將所述第一參考數據與預設閾值進行比對；

在所述第一參考數據不大于所述預設閾值時，生成第二參考數據；

根據所述第二參考數據所在的第一數值區間，選定與所述第一數值區間對應的添加對象子集合。

3.根據權利要求2所述的方法，其特征在于，所述在添加對象集合中，隨機選定添加對象子集合，還包括：

在所述第一參考數據大于所述預設閾值時，將空集合作為所述添加對象子集合。

4.根據權利要求1所述的方法，其特征在于，所述在添加對象集合中，隨機選定添加對象子集合，包括：

隨機生成第三參考數據；

根據所述第三參考數據所在的第二數值區間，選定與所述第二數值區間對應的添加對象子集合。

5.根據權利要求1至4中任意一項所述的方法，其特征在于，所述根據所述添加對象子集合，得到目標添加對象，包括：

在所述添加對象子集合為非空集合的情況下，從所述添加對象子集合中隨機選擇添加對象，作為所述目標添加對象；

在所述添加對象子集合為空集合的情況下，結束對所述原始訓練圖像進行圖像處理，將所述原始訓練圖像作為所述訓練圖像。

6.根據權利要求1至5中任意一項所述的方法，其特征在于，所述根據所述目標添加對象，對所述原始訓練圖像進行圖像處理，得到所述訓練圖像，包括：

對所述原始訓練圖像進行校正，得到校正訓練圖像；

根據所述目標添加對象，對所述校正訓練圖像進行圖像處理，得到所述訓練圖像。

7.根據權利要求6所述的方法，其特征在于，所述對所述原始訓練圖像進行校正，得到校正訓練圖像，包括：

獲取標準圖像；

對所述原始訓練圖像進行特征點提取，得到至少一個特征點，所述特征點包括左眼瞳孔特征點、右眼瞳孔特征點、鼻尖特征點、左嘴角特征點以及右嘴角特征點中的一個或兩個以上；

根據所述特征點和所述標準圖像，對所述原始訓練圖像進行仿射變換，得到校正訓練圖像。

8.一種訓練圖像的生成裝置，其特征在于，包括：

原始訓練圖像獲取模塊，用于獲取原始訓練圖像；

選定模塊，用于在添加對象集合中，隨機選定添加對象子集合；

目標添加對象獲取模塊，用于根據所述添加對象子集合，得到目標添加對象；

訓練圖像獲取模塊，用于根據所述目標添加對象，對所述原始訓練圖像進行圖像處理，得到所述訓練圖像。

9.一種電子設備，其特征在于，包括：

處理器；

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

其中，所述處理器被配置為調用所述存儲器存儲的指令，以執行權利要求1至7中任意一項所述的方法。

10.一種計算機可讀存儲介質，其上存儲有計算機程序指令，其特征在于，所述計算機程序指令被處理器執行時實現權利要求1至7中任意一項所述的方法。