[發(fā)明專利]一種圖像處理方法、裝置和電子設(shè)備

在審

• 申請?zhí)枺?/td>: 201910380522.1
• 申請日: 2019-05-08
• 公開（公告）號: CN110136052A
• 公開（公告）日: 2019-08-16
• 發(fā)明（設(shè)計）人: 李樹槐;李通;劉文韜;錢晨
• 申請（專利權(quán)）人: 北京市商湯科技開發(fā)有限公司
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06T11/40
• 代理公司: 北京派特恩知識產(chǎn)權(quán)代理有限公司 11270
• 代理人: 崔曉嵐;張穎玲

本發(fā)明實施例提供一種圖像處理方法、裝置和電子設(shè)備。所述方法包括：獲得第一圖像，所述第一圖像包括對應于待填補區(qū)域的第一目標區(qū)域以及除所述待填補區(qū)域以外的其他區(qū)域的第二目標區(qū)域；識別所述第二目標區(qū)域中的強結(jié)構(gòu)，基于所述強結(jié)構(gòu)生成強結(jié)構(gòu)補全圖像；基于所述第二目標區(qū)域以及所述強結(jié)構(gòu)補全圖像對所述第一目標區(qū)域進行填充處理，獲得第二圖像。

技術(shù)領(lǐng)域

本發(fā)明涉及圖像處理技術(shù)，具體涉及一種圖像處理方法、裝置和電子設(shè)備。

背景技術(shù)

隨著互聯(lián)網(wǎng)技術(shù)的飛速發(fā)展，出現(xiàn)了各種圖像處理工具，能夠?qū)D像中的目標對象(例如人物)進行處理，例如去除圖像中的某個目標對象。但這種處理方式雖然能夠去除圖像中不想保留的目標對象，但目標對象被去除后會在原區(qū)域留下與圖像格格不入的空白。

基于此，如何在目標對象從圖像中移除的場景下填補空白區(qū)域，目前尚無有效解決方案。

發(fā)明內(nèi)容

為解決現(xiàn)有存在的技術(shù)問題，本發(fā)明實施例提供一種圖像處理方法、裝置和電子設(shè)備。

為達到上述目的，本發(fā)明實施例的技術(shù)方案是這樣實現(xiàn)的：

本發(fā)明實施例提供了一種圖像處理方法，所述方法包括：

獲得第一圖像，所述第一圖像包括對應于待填補區(qū)域的第一目標區(qū)域以及除所述待填補區(qū)域以外的其他區(qū)域的第二目標區(qū)域；

識別所述第二目標區(qū)域中的強結(jié)構(gòu)，基于所述強結(jié)構(gòu)生成強結(jié)構(gòu)補全圖像；

基于所述第二目標區(qū)域以及所述強結(jié)構(gòu)補全圖像對所述第一目標區(qū)域進行填充處理，獲得第二圖像。

上述方案中，所述識別所述第二目標區(qū)域中的強結(jié)構(gòu)，基于所述強結(jié)構(gòu)生成強結(jié)構(gòu)補全圖像，包括：

識別所述第二目標區(qū)域中的多個強結(jié)構(gòu)，確定所述多個強結(jié)構(gòu)中滿足預設(shè)條件的目標強結(jié)構(gòu)，基于所述目標強結(jié)構(gòu)生成強結(jié)構(gòu)補全圖像。

上述方案中，所述確定所述多個強結(jié)構(gòu)中滿足預設(shè)條件的目標強結(jié)構(gòu)，包括：確定所述多個強結(jié)構(gòu)中至少部分強結(jié)構(gòu)的長度以及所述至少部分強結(jié)構(gòu)與所述第一目標區(qū)域的距離，獲得所述長度超過第一預設(shè)閾值以及所述距離小于第二預設(shè)閾值的多個目標強結(jié)構(gòu)。

上述方案中，所述基于所述目標強結(jié)構(gòu)生成強結(jié)構(gòu)補全圖像，包括：確定多個目標強結(jié)構(gòu)中至少部分目標強結(jié)構(gòu)與特定方向的夾角，基于所述夾角對所述至少部分目標強結(jié)構(gòu)進行分組，獲得強結(jié)構(gòu)分組，所述強結(jié)構(gòu)分組中包括至少一個目標強結(jié)構(gòu)；

基于所述強結(jié)構(gòu)分組中的目標強結(jié)構(gòu)的夾角和對應的位置生成強結(jié)構(gòu)補全圖像。

上述方案中，所述方法還包括：對所述第二圖像中填充處理后的所述第一目標區(qū)域進行顏色修正處理，獲得第三圖像。

上述方案中，所述對填充處理后的所述第一目標區(qū)域進行顏色修正處理，獲得第三圖像，包括：確定第三目標區(qū)域，所述第三目標區(qū)域為所述第二目標區(qū)域中貼近所述第一目標區(qū)域的區(qū)域；

對所述第一圖像和所述第二圖像中對應于所述第三目標區(qū)域的圖像進行最小二乘曲線擬合處理，獲得擬合曲線；

基于所述擬合曲線對所述第二圖像進行曲線變換，獲得第三圖像。

上述方案中，所述基于所述第二目標區(qū)域以及所述強結(jié)構(gòu)補全圖像對所述第一目標區(qū)域進行填充處理，獲得第二圖像，包括：將所述第一圖像和所述強結(jié)構(gòu)補全圖像輸入生成對抗網(wǎng)絡，獲得第二圖像，以利用所述生成對抗網(wǎng)絡對所述第一圖像中的所述第一目標區(qū)域進行填充處理。

上述方案中，所述生成對抗網(wǎng)絡包括生成器和判別器；所述生成對抗網(wǎng)絡的訓練方法包括：獲得第一樣本圖像，所述第一樣本圖像包括待填補區(qū)域；所述待填補區(qū)域的尺寸為隨機尺寸；