[發明專利]一種樣本生成方法、裝置、電子設備及計算機可讀存儲介質

在審

• 申請號: 201910969923.0
• 申請日: 2019-10-12
• 公開（公告）號: CN110889824A
• 公開（公告）日: 2020-03-17
• 發明（設計）人: 蘇睿
• 申請（專利權）人: 北京海益同展信息科技有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T5/50
• 代理公司: 北京華夏泰和知識產權代理有限公司 11662
• 代理人: 李雪

1.一種樣本生成方法，其特征在于，包括：

獲取包含豬體的初始樣本圖像及多個背景圖像；

在所述初始樣本圖像中提取豬體掩膜圖像；

對所述豬體掩膜圖像進行前景增強，得到多個增強豬體掩膜圖像；

將每個所述增強豬體掩膜圖像分別與多個所述背景圖像進行融合處理，得到多個訓練樣本圖像及標注信息。

2.根據權利要求1所述的樣本生成方法，其特征在于，所述將每個所述增強豬體掩膜圖像分別與多個所述背景圖像進行融合處理，得到多個訓練樣本圖像及標注信息，包括：

在將每個所述增強豬體掩膜圖像與任一背景圖像融合時，隨機生成所述增強豬體掩膜圖像在所述背景圖像中的添加位置；

將所述增強豬體掩膜圖像按照所述添加位置覆蓋在所述背景圖像上；

將所述增強豬體掩膜圖像與所述背景圖像進行圖像融合，得到訓練樣本圖像；

根據所述添加位置生成所述標注信息。

3.根據權利要求2所述的樣本生成方法，其特征在于，將所述增強豬體掩膜圖像按照所述添加位置覆蓋在所述背景圖像上，包括：

獲取所述增強豬體掩膜圖像的圖像尺寸；

根據所述添加位置及所述增強豬體掩膜圖像的圖像尺寸計算添加區域；

將所述增強豬體掩膜圖像添加至所述添加區域中。

4.根據權利要求3所述的樣本生成方法，其特征在于，將所述增強豬體掩膜圖像與所述背景圖像進行圖像融合，得到訓練樣本圖像，包括：

根據所述增強豬體掩膜圖像每個像素點的像素值計算所述增強豬體掩膜圖像的第一梯度場；

根據所述背景圖像中除所述添加區域外的每個像素點的計算所述背景圖像中除所述添加區域外的區域圖像的第二梯度場；

根據所述第一梯度場和所述第二梯度場計算待重建圖像的梯度場；

根據所述待重建圖像的梯度場計算散度；

根據所述散度、預設的系數矩陣和泊松方程計算訓練樣本圖像的每個像素點的RGB值。

5.根據權利要求2所述的樣本生成方法，其特征在于，根據所述添加位置生成所述標注信息，包括：

獲取所述增強豬體掩膜圖像的圖像尺寸；

根據所述添加位置及所述圖像尺寸確定標注范圍；

根據所述標注范圍生成所述標注信息。

6.根據權利要求1所述的樣本生成方法，其特征在于，前景增強的方式包括：平移、旋轉、縮放、水平翻轉、垂直翻轉、模糊處理和添加椒鹽噪聲中的一種或多種。

7.一種樣本生成裝置，其特征在于，包括：

獲取模塊，用于獲取包含豬體的初始樣本圖像及多個背景圖像；

提取模塊，用于在所述初始樣本圖像中提取豬體掩膜圖像；

前景增強模塊，用于對所述豬體掩膜圖像進行前景增強，得到多個增強豬體掩膜圖像；

融合處理模塊，用于將每個所述增強豬體掩膜圖像分別與多個所述背景圖像進行融合處理，得到多個訓練樣本圖像及標注信息。

8.根據權利要求7所述的樣本生成裝置，其特征在于，所述融合處理模塊，包括：

位置生成單元，用于在將每個所述增強豬體掩膜圖像與任一背景圖像融合時，隨機生成所述增強豬體掩膜圖像在所述背景圖像中的添加位置；

覆蓋單元，用于將所述增強豬體掩膜圖像按照所述添加位置覆蓋在所述背景圖像上；

融合單元，用于將所述增強豬體掩膜圖像與所述背景圖像進行圖像融合，得到訓練樣本圖像；

生成單元，用于根據所述添加位置生成所述標注信息。

9.根據權利要求8所述的樣本生成裝置，其特征在于，所述覆蓋單元，包括：

第一獲取子單元，用于獲取所述增強豬體掩膜圖像的圖像尺寸；

第一計算子單元，用于根據所述添加位置及所述增強豬體掩膜圖像的圖像尺寸計算添加區域；

添加子單元，用于將所述增強豬體掩膜圖像添加至所述添加區域中。