[發明專利]基于二維高斯分布的圖像數據增強方法、裝置及存儲介質

有效

• 申請號: 202110610001.8
• 申請日: 2021-06-01
• 公開（公告）號: CN113222873B
• 公開（公告）日: 2023-06-16
• 發明（設計）人: 韋嘉楠;周超勇;劉玉宇
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: G06T5/50
• 分類號: G06T5/50
• 代理公司: 北京鴻元知識產權代理有限公司 11327
• 代理人: 王迎;袁文婷

1.一種基于二維高斯分布的圖像數據增強方法，其特征在于，所述方法包括：

獲取待增強樣本數據中的任意兩張圖像，形成待測圖像和隨機圖像，并獲取所述待測圖像和所述隨機圖像的二維高斯分布；

根據所述待測圖像和所述隨機圖像的二維高斯分布，確定與所述待測圖像和所述隨機圖像分別對應的掩碼矩陣；

基于所述待測圖像和所述隨機圖像的掩碼矩陣，確定所述待測圖像和所述隨機圖像的結合圖像；

對所述結合圖像中的所述待測圖像的邊界框和所述隨機圖像的邊界框進行組合，確定與所述結合圖像對應的邊界框信息；

基于所述結合圖像及所述邊界框信息，確定與所述待測圖像和所述隨機圖像對應的增強圖像，以增強所述待增強樣本數據；

其中，所述待測圖像和所述隨機圖像存儲于區塊鏈中，所述獲取所述待測圖像和所述隨機圖像的二維高斯分布，包括：

在所述待測圖像和所述隨機圖像上分別隨機生成與所述待測圖像和所述隨機圖像分別對應的中心點；

基于所述待測圖像和所述隨機圖像的中心點，分別生成與所述待測圖像和所述隨機圖像相對應的二維高斯分布；

所述待測圖像的中心點表示為(x₁,y₁)，所述隨機圖像的中心點表示為(x₂,y₂)，所述待測圖像的高斯分布表示為N₁(x₁，y₁，1，1，ρ₁)，所述隨機圖像的高斯分布表示為N₂(x₂，y₂，1，1，ρ₂)；其中，

所述待測圖像在像素點為(m，n)處的概率密度表示為：

所述隨機圖像的在像素點(u，v)處的概率密度表示為：

其中，σ₁²＝1，σ₂²＝1，ρ₁表示二維高斯分布N₁中正態隨機變量x₁和正態隨機變量y₁之間的相關系數，取值范圍為-1ρ₁1，ρ₂表示二維高斯分布N₂正態隨機變量x₂和正態隨機變量y₂之間的相關系數，取值范圍為-1＜ρ₂＜1。

2.如權利要求1所述的基于二維高斯分布的圖像數據增強方法，其特征在于，所述確定與所述待測圖像和所述隨機圖像分別對應的掩碼矩陣，包括：

基于所述待測圖像的二維高斯分布，確定所述待測圖像中每個像素點對應的第一概率密度，基于所述隨機圖像的二維高斯分布，確定所述隨機圖像中每個像素點對應的第二概率密度；

基于所述第一概率密度確定所述待測圖像的第一掩碼矩陣，基于所述第二概率密度，確定所述隨機圖像的第二掩碼矩陣。

3.如權利要求2所述的基于二維高斯分布的圖像數據增強方法，其特征在于，所述確定所述待測圖像和所述隨機圖像的結合圖像，包括：

基于所述第一掩碼矩陣和所述第二掩碼矩陣，確定所述待測圖像和所述隨機圖像的逐像素乘法圖像；

基于所述待測圖像和所述隨機圖像的逐像素乘法圖像之和，確定所述結合圖像。

4.如權利要求3所述的基于二維高斯分布的圖像數據增強方法，其特征在于，所述基于所述第一概率密度確定所述待測圖像的第一掩碼矩陣，基于所述第二概率密度，確定所述隨機圖像的第二掩碼矩陣，包括：

所述第一掩碼矩陣表示為：

所述第二掩碼矩陣表示為：

其中，f₁(m，n)表示所述待測圖像在像素點為(m，n)處的概率密度，f₁(u，v)表示所述隨機圖像在像素點(u，v)處的概率密度。