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 该库,多谢!

原文发布于微信公众号 - AI研习社(okweiwu)

原文发表时间:2018-05-16

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏数据小魔方

创意滑珠图!

今天要给大家分享的是一种非常有趣的滑珠图! ▽ 本文要讲解的滑珠图做法,稍微有点复杂。不过这种滑珠图在数据对比展示中,效果奇佳。小魔方参考多处教程和资料,终于还...

2864
来自专栏1007261的专栏

基于图像识别的自动化

导语 在客户端自动化中,如果需要对UI进行操作,控件识别和操作是最基础的能力。在windows标准控件中,我们可以通过FindWindow来找到窗口,FindW...

2.7K7
来自专栏专知

【下载】TensorFlow机器学习教程手把手书谱

【导读】 来自 Nick McClure 的电子书《TensorFlow Machine Learning Cookbook》这本书主要是基于TensorFlo...

3778
来自专栏机器之心

教程 | PyTorch经验指南:技巧与陷阱

项目地址:https://github.com/Kaixhin/grokking-pytorch

4692
来自专栏机器之心

教程 | 用摄像头和Tensorflow.js在浏览器上实现目标检测

3214
来自专栏一心无二用,本人只专注于基础图像算法的实现与优化。

PhotoShop算法原理解析系列 - 风格化---》查找边缘。                  闲谈.Net类型之public的不public,fixed的不能fixed     当然这个还可

      之所以不写系列文章一、系列文章二这样的标题,是因为我不知道我能坚持多久。我知道我对事情的表达能力和语言的丰富性方面的天赋不高。而一段代码需要我去用心...

3149
来自专栏iOSDevLog

quickdraw_datasetQuick Draw!数据集

https://console.cloud.google.com/storage/browser/quickdraw_dataset

4912
来自专栏奇点大数据

Pytorch神器(4)

上一次,我们用最简短的篇幅讲述了用Pytorch实现线性回归的过程。整个程序仅仅用了约60多行就完成了一个线性回归机器学习程序的全部内容。这次的文章,我们来对上...

1123
来自专栏Coding迪斯尼

vue+webpack实现精美游戏设计:实现建筑物的渐变生成效果

1043
来自专栏量子位

谷歌推出Tangent开源库,在Python源代码上做自动微分

李林 编译自 Google Research Blog 量子位 出品 | 公众号 QbitAI 谷歌今天推出了一个新的开源Python自动微分库:Tangent...

3715

扫码关注云+社区

领取腾讯云代金券