天天看點

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

1. 摘要

作者提出了一種生成式圖像修複系統,該系統基于從數百萬個圖像中學習的門控卷積,無需額外的标記工作。作者所提出的卷積解決了将所有輸入像素都視為有效像素的香草卷積問題,通過為所有通道在所有層上的每個空間位置提供可學習的動态特征選擇機制來概括部分卷積。

此外,由于自由形式的蒙版可能會出現在任何形狀的圖像中,是以為單個矩形蒙版設計的全局和局部GAN均不适用。是以,我們還通過應用頻譜歸一化提出了基于更新檔的GAN損失,稱為SN-PatchGAN鑒别密集圖像斑塊。SN-PatchGAN的配方簡單,訓練快速,穩定。自動圖像修複和使用者指導的擴充的結果表明,與以前的方法相比,我們的系統可産生更高品質和更靈活的結果。我們的系統可幫助使用者迅速移除分散注意力的物體,修改圖像布局,清除水印并編輯臉部。

下圖顯示處理:

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後
圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

作者在項目中提供了一個互動式示範,我們可以自由地遮蓋圖像的某些部分,然後檢查其生成效果。Deepfill V2提供了兩個模型,這些模型在兩個資料集上進行了預訓練:places2和celebahq。從效果的角度來看,至少對于這兩個資料集,它在場景和面部圖像中做得非常好,尤其是在人臉補全效果突出。

作者在項目中提供了一個互動式示範,我們可以自由地遮蓋圖像的某些部分,然後檢查其生成效果。Deepfill V2提供了兩個模型,這些模型在兩個資料集上進行了預訓練:places2和celebahq。從效果的角度來看,至少對于這兩個資料集,它在場景和面部圖像中做得非常好,尤其是在人臉補全效果突出。

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

第一行是真實圖像,第二行是删除細節的圖像,第三行是DeepFill v2修複的圖像

作者指出,該項目隻有三個依賴項,即Python 3、TensorFlow和他制作的TF工具包neuralgym。作者在TF 1.3、1.4、1.5、1.6、1.7上進行了測試,并将各種模型超參數放入YML檔案中,以便于調整。

2. 網絡結構

DeepFill,是基于一個門控卷積網絡,能夠處理任意形式的掩膜,以及使用者手繪的線條。這個深度學習系統可以幫助使用者快速擦除畫面上不想要的部分,修改圖像的布局、編輯面部,以及在圖中互動式建立新對象。

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

上圖左邊是局部卷積網絡,右邊是門控卷積網絡。

對于這個神經網絡而言,輸入資訊分成三個:顔色通道(RGB Channel)、掩膜通道(Mask Channel)、手繪通道(Sketch Channel)。

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

當然除了門控卷積網絡之外,DeepFill裡還包括SN-PatchGAN(光譜歸一化馬爾可夫判别器)等關鍵部件。

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

3. 測試

我們來看下作者針對不同方法的比較,包括PatchMatch,Global&Local,ContextAttention,PartialConv和我們的方法。圖檔修複的比較基于四個次元:Semantics,Non-Local,Free-Form和User-guided選項如下圖所示:

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

均值誤差比較:

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

下面我們來看下它的修複效果:

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

上圖最左邊是輸入,照片被各種塗抹,白色的部分就是所謂的掩膜,黑色的細線就是手繪的部分。中間是各種神經網絡的處理結果,其中标注Ours的圖檔,就是DeepFill的處理結果。

處理修複圖檔之外,還可以對圖檔進行編輯:

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

上圖,按照想要的方式,修剪遠處的小樹叢。

顯然,DeepFill确實厲害。

4. 最後

圖像修複之DeepFill: Free-Form Image Inpainting with Gated Convolution1. 摘要2. 網絡結構3. 測試4. 最後

繼續閱讀