用PyTorch 實現全局/局部一致圖像補全

本庫用 PyTorch 實現了全局/局部一致圖像補全（Globally and Locally Consistent Image Completion ）。

論文鏈接：

http://hi.cs.waseda.ac.jp/~iizuka/projects/completion/en/

Github：

https://github.com/akmtn/pytorch-siggraph2017-inpainting

結果：

gif: input -> masked -> output -> blackout

result1

result2

result3

Requirements

Python 3

PyTorch

NumPy

OpenCV

pyamg（可選，但需要進行後期處理）

scipy（可選，但需要進行後期處理）

這段代碼不用 GPU 可以運行

下載預訓練模型：

wget --continue-O completionnet_places2.t7 http://hi.cs.waseda.ac.jp/~iizuka/data/completionnet_places2.t7

用法

基礎用法

pythoninpaint.py--input--mask

下面的 mask 是一張黑白圖片，白色完全是 reagion，並且 mask 的大小應該和輸入的圖像一樣。如果 mask 沒有指定，則會隨機生成 mask。

其他選項：

--gpu：用 GPU 計算，可能不會變快。 默認為 False。

--postproc：可以進行後期處理（泊松融合）。 默認為 False，這個泊松融合腳本是從這個（https://github.com/parosky/poissonblending）庫分出的。

例子：

pythoninpaint.py--inputexample.png --mask example_mask.png

請star該庫，多謝！

從Python入門-如何成為AI工程師

BAT資深演算法工程師獨家研發課程

最貼近生活與工作的好玩實操項目

班級管理助學搭配專業的助教答疑

學以致用拿offer，學完即推薦就業

新人福利

關注 AI 研習社（okweiwu），回復1領取

【超過 1000G 神經網路 / AI / 大數據資料】

DeepMind 新研究：使用強化對抗學習合成圖像程序

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！