[發(fā)明專利]圖像去霧模型的生成方法、裝置及圖像去霧方法、裝置

在審

• 申請?zhí)枺?/td>: 201910209125.8
• 申請日: 2019-03-19
• 公開（公告）號: CN109949242A
• 公開（公告）日: 2019-06-28
• 發(fā)明（設(shè)計）人: 房建東;趙肖肖;趙于東
• 申請（專利權(quán)）人: 內(nèi)蒙古工業(yè)大學(xué)
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/50
• 代理公司: 北京金信知識產(chǎn)權(quán)代理有限公司 11225
• 代理人: 喻嶸;郭迎俠

1.一種圖像去霧模型的生成方法，其特征在于，所述方法包括：

獲取訓(xùn)練數(shù)據(jù)，其中，所述訓(xùn)練數(shù)據(jù)包括樣本有霧圖像、與樣本有霧圖像對應(yīng)的樣本無霧圖像以及樣本無霧圖像的標(biāo)識信息；

執(zhí)行以下訓(xùn)練步驟：

將所述訓(xùn)練數(shù)據(jù)中的樣本有霧圖像輸入條件生成式對抗網(wǎng)絡(luò)模型中的生成模型，生成與輸入的樣本有霧圖像對應(yīng)的無霧圖像；根據(jù)生成的無霧圖像和目標(biāo)樣本無霧圖像，確定生成模型的第一損失函數(shù)，其中，所述目標(biāo)樣本無霧圖像為與輸入生成模型的樣本有霧圖像對應(yīng)的樣本無霧圖像；將目標(biāo)樣本無霧圖像和生成模型生成的無霧圖像輸入條件生成式對抗網(wǎng)絡(luò)模型中的判別模型，輸出圖像判別結(jié)果；根據(jù)圖像判別結(jié)果和目標(biāo)樣本無霧圖像的標(biāo)識信息，確定判別模型的第二損失函數(shù)；根據(jù)第一損失函數(shù)和第二損失函數(shù)，統(tǒng)計條件生成式對抗網(wǎng)絡(luò)模型的損失函數(shù)，并確定損失函數(shù)是否達(dá)到預(yù)設(shè)閾值；響應(yīng)于確定損失函數(shù)達(dá)到預(yù)設(shè)閾值，根據(jù)當(dāng)前訓(xùn)練后的條件生成式對抗網(wǎng)絡(luò)模型生成圖像去霧模型，以及輸出所述圖像去霧模型。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述獲取訓(xùn)練數(shù)據(jù)，包括：

采集目標(biāo)地區(qū)的遙感數(shù)據(jù)集，其中，所述遙感數(shù)據(jù)集中包含至少一種波段的遙感數(shù)據(jù)；

根據(jù)所述遙感數(shù)據(jù)集中的遙感數(shù)據(jù)，生成訓(xùn)練數(shù)據(jù)。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述根據(jù)所述遙感數(shù)據(jù)集中的遙感數(shù)據(jù)，生成訓(xùn)練數(shù)據(jù)，包括：

將所述遙感數(shù)據(jù)集中的至少一種波段的遙感數(shù)據(jù)進(jìn)行合成，生成合成圖像，其中，所述合成圖像包括有霧合成圖像和無霧合成圖像；

根據(jù)生成的合成圖像，生成訓(xùn)練數(shù)據(jù)。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述將所述遙感數(shù)據(jù)集中的至少一種波段的遙感數(shù)據(jù)進(jìn)行合成，包括以下至少一項：

將至少兩種波段的有霧遙感數(shù)據(jù)進(jìn)行合成；或者

將至少兩種波段的無霧遙感數(shù)據(jù)進(jìn)行合成；或者

將至少一種波段的有霧圖像與至少一種波段的無霧圖像進(jìn)行合成。

5.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述根據(jù)生成的合成圖像，生成訓(xùn)練數(shù)據(jù)，包括：

對生成的合成圖像進(jìn)行裁剪，得到多張預(yù)設(shè)尺寸的裁剪圖像；

根據(jù)得到的裁剪圖像，生成訓(xùn)練數(shù)據(jù)。

6.根據(jù)權(quán)利要求1-5之一所述的方法，其特征在于，所述方法還包括：

響應(yīng)于確定損失函數(shù)未達(dá)到預(yù)設(shè)閾值，對當(dāng)前訓(xùn)練后的條件生成式對抗網(wǎng)絡(luò)模型進(jìn)行優(yōu)化；將優(yōu)化后的模型作為條件生成式對抗網(wǎng)絡(luò)模型，并繼續(xù)執(zhí)行所述訓(xùn)練步驟。

7.一種圖像去霧模型的生成裝置，其特征在于，所述裝置包括：

獲取單元，被配置成獲取訓(xùn)練數(shù)據(jù)，其中，所述訓(xùn)練數(shù)據(jù)包括樣本有霧圖像、與樣本有霧圖像對應(yīng)的樣本無霧圖像以及樣本無霧圖像的標(biāo)識信息；

執(zhí)行單元，被配置成執(zhí)行以下訓(xùn)練步驟：

將所述訓(xùn)練數(shù)據(jù)中的樣本有霧圖像輸入條件生成式對抗網(wǎng)絡(luò)模型中的生成模型，生成與輸入的樣本有霧圖像對應(yīng)的無霧圖像；根據(jù)生成的無霧圖像和目標(biāo)樣本無霧圖像，確定生成模型的第一損失函數(shù)，其中，所述目標(biāo)樣本無霧圖像為與輸入生成模型的樣本有霧圖像對應(yīng)的樣本無霧圖像；將目標(biāo)樣本無霧圖像和生成模型生成的無霧圖像輸入條件生成式對抗網(wǎng)絡(luò)模型中的判別模型，輸出圖像判別結(jié)果；根據(jù)圖像判別結(jié)果和目標(biāo)樣本無霧圖像的標(biāo)識信息，確定判別模型的第二損失函數(shù)；根據(jù)第一損失函數(shù)和第二損失函數(shù)，統(tǒng)計條件生成式對抗網(wǎng)絡(luò)模型的損失函數(shù)，并確定損失函數(shù)是否達(dá)到預(yù)設(shè)閾值；響應(yīng)于確定損失函數(shù)達(dá)到預(yù)設(shè)閾值，根據(jù)當(dāng)前訓(xùn)練后的條件生成式對抗網(wǎng)絡(luò)模型生成圖像去霧模型，以及輸出所述圖像去霧模型。

8.根據(jù)權(quán)利要求7所述的裝置，其特征在于，所述獲取單元包括：

采集子單元，被配置成采集目標(biāo)地區(qū)的遙感數(shù)據(jù)集，其中，所述遙感數(shù)據(jù)集中包含至少一種波段的遙感數(shù)據(jù)；

生成子單元，被配置成根據(jù)所述遙感數(shù)據(jù)集中的遙感數(shù)據(jù)，生成訓(xùn)練數(shù)據(jù)。