[發(fā)明專利]一種基于對抗網(wǎng)絡(luò)的道路箭頭圖片生成方法及系統(tǒng)

在審

• 申請?zhí)枺?/td>: 202110868084.0
• 申請日: 2021-07-30
• 公開（公告）號: CN113780068A
• 公開（公告）日: 2021-12-10
• 發(fā)明（設(shè)計）人: 何云;何豪杰;羅躍軍
• 申請（專利權(quán)）人: 武漢中海庭數(shù)據(jù)技術(shù)有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 武漢藍寶石專利代理事務(wù)所(特殊普通合伙) 42242
• 代理人: 朱才永

1.一種基于對抗網(wǎng)絡(luò)的道路箭頭圖片生成方法，其特征在于，包括：

向預(yù)先訓(xùn)練后的對抗網(wǎng)絡(luò)模型中分別輸入目標(biāo)圖片和參考圖片，獲取所述對抗網(wǎng)絡(luò)模型輸出的與所述目標(biāo)箭頭的類型相同，且與所述參考箭頭的風(fēng)格信息和背景信息相同的新目標(biāo)箭頭，其中，所述目標(biāo)圖片中包括目標(biāo)箭頭，所述參考圖片中包括參考箭頭；

將新目標(biāo)箭頭融合到所述參考圖片中。

2.根據(jù)權(quán)利要求1所述的道路箭頭圖片生成方法，其特征在于，所述對抗網(wǎng)絡(luò)模型包括生成器G、映射網(wǎng)絡(luò)M、風(fēng)格編碼器E和鑒別器D，對抗網(wǎng)絡(luò)模型的訓(xùn)練過程包括：

收集不同場景下的道路圖像，提取每一張道路圖像中的道路箭頭區(qū)域，并對道路箭頭的類型進行標(biāo)注，作為訓(xùn)練樣本集，對所述對抗網(wǎng)絡(luò)模型進行訓(xùn)練。

3.根據(jù)權(quán)利要求2所述的道路箭頭圖片生成方法，其特征在于，所述對所述對抗網(wǎng)絡(luò)模型進行訓(xùn)練，包括：

向生成器G中輸入道路箭頭的目標(biāo)類型和原始圖片，所述原始圖片中道路箭頭的類型與道路箭頭的目標(biāo)類型不同，獲取所述生成器G生成的第一新圖片，其中，第一新圖片中的道路箭頭的類型為目標(biāo)類型；

將生成的第一新圖片分別輸入判別器D和再次回環(huán)輸入生成器G中，通過判別器D判別生成的第一新圖片為虛假圖片還是真實圖片，以及第一新圖片中道路箭頭的類型，通過再次輸入生成器G中，生成第二新圖片，所述第二新圖片中的道路箭頭的類型與原始圖片中道路箭頭的類型相同；

計算第二新圖片中的道路箭頭和原始圖片中的道路箭頭之間的損失，反復(fù)訓(xùn)練得到最優(yōu)的對抗網(wǎng)絡(luò)模型。

4.根據(jù)權(quán)利要求3所述的道路箭頭圖片生成方法，其特征在于，所述第二新圖片中的道路箭頭和原始圖片中的道路箭頭之間的損失包括對抗損失、風(fēng)格重建損失、風(fēng)格差別多元化損失和循環(huán)一致性損失。

5.根據(jù)權(quán)利要求1-3任一項所述的道路箭頭圖片生成方法，其特征在于，所述道路箭頭的類型包括直行箭頭和左轉(zhuǎn)箭頭，所述目標(biāo)圖片中目標(biāo)箭頭的類型為左轉(zhuǎn)箭頭，所述參考圖片中參考箭頭的類型為直行箭頭，所述生成的新目標(biāo)箭頭為新左轉(zhuǎn)箭頭。

6.根據(jù)權(quán)利要求5所述的道路箭頭圖片生成方法，其特征在于，所述風(fēng)格信息和背景信息至少包括導(dǎo)向箭頭的傾斜角度、大小、圖片的對比度和亮度。

7.根據(jù)權(quán)利要求6所述的道路箭頭圖片生成方法，其特征在于，所述將新目標(biāo)箭頭融合到所述參考圖片中，包括：

調(diào)整生成的新左轉(zhuǎn)箭頭的亮度和對比度與參考圖片中直行箭頭的亮度和對比度一致；

基于泊松融合算法調(diào)整新左轉(zhuǎn)箭頭的風(fēng)格和背景顏色，獲取調(diào)整后的新左轉(zhuǎn)箭頭；

將調(diào)整后的新左轉(zhuǎn)箭頭覆蓋參考圖片中的直行箭頭，生成最終的道路箭頭圖片。

8.一種基于對抗網(wǎng)絡(luò)的道路箭頭圖片生成系統(tǒng)，其特征在于，包括：

輸入模塊，用于向預(yù)先訓(xùn)練后的對抗網(wǎng)絡(luò)模型中分別輸入目標(biāo)圖片和參考圖片，所述目標(biāo)圖片中包括目標(biāo)箭頭，所述參考圖片中包括參考箭頭；

獲取模塊，用于獲取所述對抗網(wǎng)絡(luò)模型輸出的與所述目標(biāo)箭頭的類型相同，且與所述參考箭頭的風(fēng)格信息和背景信息相同的新目標(biāo)箭頭；

融合模塊，用于將新目標(biāo)箭頭融合到所述參考圖片中。

9.一種電子設(shè)備，其特征在于，包括存儲器、處理器，所述處理器用于執(zhí)行存儲器中存儲的計算機管理類程序時實現(xiàn)如權(quán)利要求1-7任一項所述的基于對抗網(wǎng)絡(luò)的道路箭頭圖片生成方法的步驟。

10.一種計算機可讀存儲介質(zhì)，其特征在于，其上存儲有計算機管理類程序，所述計算機管理類程序被處理器執(zhí)行時實現(xiàn)如權(quán)利要求1-7任一項所述的基于對抗網(wǎng)絡(luò)的道路箭頭圖片生成方法的步驟。