本库用 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
这段代码不用 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。
其他选项:
例子:
python inpaint.py --input example.png --mask example_mask.png
请 star 该库,多谢!