[發明專利]一種二維碼訓練數據集的構建方法及裝置

在審

• 申請號: 201911351968.8
• 申請日: 2019-12-25
• 公開（公告）號: CN111160501A
• 公開（公告）日: 2020-05-15
• 發明（設計）人: 張冰洋;姚寒星
• 申請（專利權）人: 北京亮亮視野科技有限公司
• 主分類號: G06K19/06
• 分類號: G06K19/06;G06T11/60
• 代理公司: 北京三友知識產權代理有限公司 11127
• 代理人: 周永君;孫乳筍

1.一種二維碼訓練數據集的構建方法，其特征在于，包括：

對二維碼圖片集中的每張二維碼圖片進行場景處理，獲得每張二維碼圖片對應的場景二維碼圖片；

將每張二維碼圖片對應的場景二維碼圖片與從背景圖片集中隨機獲取的一張背景圖片進行合成，獲得二維碼訓練數據集的二維碼訓練圖片。

2.根據權利要求1所述的方法，其特征在于，所述對二維碼圖片集中的每張二維碼圖片進行場景處理，獲得每張二維碼圖片對應的場景二維碼圖片包括：

對每張二維碼圖片進行隨機并排處理、相似變換處理和隨機縮放處理中的至少一種處理，獲得每張二維碼圖片對應的場景二維碼圖片。

3.根據權利要求2所述的方法，其特征在于，所述對每張二維碼圖片進行隨機并排處理包括：

以第一概率保留所述二維碼圖片本身，以第二概率將所述二維碼圖片與第一預設張數的二維碼圖片單排橫向排列，以第三概率將所述二維碼圖片與第二預設張數的二維碼圖片單列縱向排列和以第四概率將所述二維碼圖片與第三預設張數的二維碼圖片多排多列排列；其中，所述第一概率、所述第二概率、所述第三概率和所述第四概率之和為1。

4.根據權利要求2所述的方法，其特征在于，所述對每張二維碼圖片進行相似變換處理包括：

通過透視變換矩陣M對每張二維碼圖片進行相似變換處理，所述透視變換矩陣M為：

其中，x和y為-0.3到0.3之間的隨機數，c₁和c₂為-0.0005到0.0005之間的隨機數。

5.根據權利要求2所述的方法，其特征在于，所述對每張二維碼圖片進行隨機縮放處理包括：

以第五概率將每張二維碼圖片的尺寸縮放至第一尺寸范圍，以第六概率將每張二維碼圖片的尺寸縮放至第二尺寸范圍，以第七概率將每張二維碼圖片的尺寸縮放至第三尺寸范圍和以第八概率將每張二維碼圖片的尺寸縮放至第四尺寸范圍；其中，所述第五概率、所述第六概率、所述第七概率和所述第八概率之和為1。

6.根據權利要求1所述的方法，其特征在于，所述將每張二維碼圖片對應的場景二維碼圖片與從背景圖片集中隨機獲取的一張背景圖片進行合成，獲得二維碼訓練數據集的二維碼訓練圖片包括：

在將每張二維碼圖片對應的場景二維碼圖片與從背景圖片集中隨機抽取的一張背景圖片進行合成之后，對每張二維碼圖片對應的合成圖片進行圖像調整處理獲得每張二維碼圖片對應的調整合成圖片；

將每張二維碼圖片對應的調整合成圖片作為所述二維碼訓練圖片。

7.根據權利要求6所述的方法，其特征在于，所述對每張二維碼圖片對應的合成圖片進行圖像調整處理獲得每張二維碼圖片對應的調整合成圖片包括：

隨機改變每張二維碼圖片對應的合成圖片的亮度、對比度、飽和度和色度中的至少一種。

8.根據權利要求1至7任一項所述的方法，其特征在于，所述將每張二維碼圖片對應的場景二維碼圖片與從背景圖片集中隨機獲取的一張背景圖片進行合成包括：

將每張二維碼圖片對應的場景二維碼圖片進行旋轉處理和/或隨機平移處理。

9.一種二維碼訓練數據集的構建裝置，其特征在于，包括：

場景處理單元，用于對二維碼圖片集中的每張二維碼圖片進行場景處理，獲得每張二維碼圖片對應的場景二維碼圖片；

合成單元，用于將每張二維碼圖片對應的場景二維碼圖片與從背景圖片集中隨機獲取的一張背景圖片進行合成，獲得二維碼訓練數據集的二維碼訓練圖片。

10.一種電子設備，包括存儲器、處理器及存儲在存儲器上并可在處理器上運行的計算機程序，其特征在于，所述處理器執行所述計算機程序時實現權利要求1至8任一項所述方法的步驟。

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