用深度神經網路修復H漫:看完這篇你就能眼中無碼

技術小能手發表於2018-10-29

a5dde00ea94dd896ec280d4842b534a1807fead5

AI“腦補”能力一流,現在甚至已經能畫出藝術品。熱愛H漫的死宅們靈光一閃,AI是否也可以把馬賽克阻擋的內容也畫出來呢?

果然,原始動物本能是第一科技生產力。最近就有人在GitHub上釋出了一個DeepCreamPy專案,能幫你把H漫中羞羞的畫面補上。

95bd69a569516a7859d3f5755533b52fd8cbb5bc

該專案使用深度完全卷積神經網路(deep fully convolutional neural network),參照了英偉達在今年4月前釋出的一篇論文。當然,英偉達原文的目的可不是用來做羞羞的事情,而是為了復原畫面被單色條帶遮擋的問題。

870abb84c97fe0dd8f1ecd6ed743bff1091adf81

從實際效果來看,復原後的圖片塗抹痕跡仍然比較明顯,不過處理線條比較簡單的漫畫可以說是綽綽有餘。

接下來,就是讓你“眼中無碼”的DIY教程啦!

適用範圍

DeepCreamPy僅適用於薄碼,如果馬賽克太大太厚,去碼可能會失效。另外,它對真人圖片無效。如果你非要嘗試,可以看一下強行使用的效果:

45a3e7cab5522b1a94a8bb8269f37d7172471810

而且DeepCreamPy目前的版本還不能完全自動處理圖片,需要用Photoshop首先對馬賽克部分進行手動預處理。

如果你十分介意這些缺點,那麼後面的內容可以跳過了。如果你學(wu)習(ma)的意願很強烈,請接著往下看。

使用方法

安裝軟體

64位Windows使用者可以直接下載exe程式,資源地址請見附錄。

如果你使用的不是Windows系統,需要自己編譯,程式中已經包含訓練模型,請從壓縮包中提取。

編譯程式碼需要一下元件:

 ●  Python 3.6
 ●  TensorFlow 1.10
 ●  Keras 2.2.4
 ●  Pillow
 ●  h5py

Tensorflow、Keras、Pillow、和h5py可以使用pip安裝:

$ pip install -r requirements.txt

請注意軟體版本,Windows上的TensorFlow不相容Python 2,也不相容Python 3.7。

手動處理黑條遮擋和馬賽克遮擋

首先開啟Photoshop或者GIMP。對於單色條遮住人物敏感部位的情況,使用純綠色(色號#00FF00#)預處理圖片,以綠條取代圖片中的黑條。

c840fe71122e13748699d18efdaa6c36ce43ef84

強烈建議使用鉛筆而不是毛刷工具,如果使用毛刷,請一定要關閉抗鋸齒功能。或者用魔棒選中馬賽克區域,再用油漆桶上色。

最後將處理的檔案以PNG格式儲存在軟體的”decensor_input”資料夾中。

如果敏感部位不是黑條,而是馬賽克,還需要將未上色的原始圖片放入”decensor_input_original” 資料夾中,並確保其檔名和放在”decensor_input”中的預處理圖片檔名相同。

執行去馬賽克軟體

A. 使用exe直接處理

雙擊軟體即可

B. 命令列處理

對於黑條遮擋的圖片,輸入以下命令:

$ python decensor.py

對於馬賽克遮擋的圖片,輸入以下命令:

$ python decensor.py —is_mosaic=True

注意事項

如果你圖片處理後成了這樣:

86dd5b1ac40beb77e0d4261975c81ea895fbd75e

一定是你處理的姿勢不對,請注意不要犯以下兩種錯誤:

4be4c30f41b9203c1e4ebd695188b2d8d0418256

第一幅圖中,圖片馬賽克區域沒有完全塗滿;第二幅圖中,由於開啟了抗鋸齒功能,導致馬賽克邊緣區域不是純綠色填充,請關閉抗鋸齒功能!!

最後給伸手黨們附上軟體地址,請自行下載,低調使用。Enjoy it!

原文釋出時間為:2018-10-29
本文作者:曉查
本文來自雲棲社群合作伙伴“量子位”,瞭解相關資訊可以關注“量子位”。


相關文章