Github 项目推荐 | 用 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

用法

基础用法

python inpaint.py --input <input_image> --mask <mask_image>

下面的 mask 是一张黑白图片，白色完全是 reagion，并且 mask 的大小应该和输入的图像一样。如果 mask 没有指定，则会随机生成 mask。

其他选项：

• --gpu：用 GPU 计算，可能不会变快。 默认为 False。
• --postproc：可以进行后期处理（泊松融合）。 默认为 False，这个泊松融合脚本是从这个（https://github.com/parosky/poissonblending）库分出的。

例子：

python inpaint.py --input example.png --mask example_mask.png

请 star 该库，多谢！