[發明專利]一種圖像數據增廣方法

在審

• 申請號: 202310403244.3
• 申請日: 2023-04-13
• 公開（公告）號: CN116664969A
• 公開（公告）日: 2023-08-29
• 發明（設計）人: 張大飛
• 申請（專利權）人: 重慶銳明信息技術有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V20/70;G06V10/82;G06N3/0475;G06N3/094
• 代理公司: 重慶強大凱創專利代理事務所(普通合伙) 50217
• 代理人: 趙玉乾

1.一種圖像數據增廣方法，其特征在于：包括：

S1：采集標準場景下的圖像，并按照預設的場景標簽進行標注，生成訓練數據集一，并獲取用于目標檢測的目標數據；

S2：基于訓練數據集一，訓練建立多場景融合參數融合的生成對抗網絡；所述生成對抗網絡包括生成器、判別器、場景映射網絡以及場景編碼器；

S3：將目標數據、預設的場景標簽取值范圍以及目標數據中圖像對應的場景標簽個數輸入至生成對抗網絡中，并預設采樣范圍和預設采樣個數輸入至生成對抗網絡，輸出增廣后的圖像數據。

2.根據權利要求1所述的一種圖像數據增廣方法，其特征在于：所述S1中預設的場景標簽包括色溫參數、亮度參數和運動模糊度參數。

3.根據權利要求2所述的一種圖像數據增廣方法，其特征在于：所述S2中包括：

S2-1：將色溫參數、亮度參數取對數后，和運動模糊度參數按照預設采樣范圍和預設采樣個數經采樣后組成掩碼向量，并和圖像一同輸入映射網絡中映射出風格碼；

S2-2：通過場景編碼器從映射網絡中提取風格碼。

4.根據權利要求3所述的一種圖像數據增廣方法，其特征在于：所述S3包括：

S3-1：將風格碼和目標數據中的圖像通過生成器得到輸出圖像；

S3-2：采用判別器，通過損失函數計算生成器的損失值，根據損失值調整對應的網絡參數，直至得到收斂的生成對抗網絡。

5.根據權利要求4所述的一種圖像數據增廣方法，其特征在于：所述S3-2中的損失函數包括對抗損失和風格多樣損失，所述對抗損失的計算方式如下：

其中，x表示圖像，m表示掩碼向量；

所述風格多樣損失計算公式如下：

其中，

6.根據權利要求5所述的一種圖像數據增廣方法，其特征在于：目標數據中的圖像數據尺寸與生成器的輸入尺寸一致。