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 条评论
登录 后参与评论

相关文章

来自专栏Coding迪斯尼

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

983
来自专栏iOSDevLog

seaborn的介绍

Seaborn是一个用Python制作统计图形的库。它建立在matplotlib之上,并与pandas数据结构紧密集成。

2631
来自专栏BestSDK

看完这个,不用写代码就能实现深度学习了

2006年,机器学习界泰斗Hinton,在Science上发表了一篇使用深度神经网络进行维数约简的论文 ,自此,神经网络再次走进人们的视野,进而引发了一场深度学...

5355
来自专栏FreeBuf

干货 | 黑客带你还原韩剧《幽灵》中出现的隐写术

0x00 前言 最近有一部剧人气非常高,据说是将军官和医生的浪漫故事。咦?!!这不是好几年前的老片子了么。 后来有人告诉我,是一部韩剧,里面的欧巴超帅妹子超靓。...

3527
来自专栏人工智能头条

张雨石:Adam - 大规模分布式机器学习框架

5457
来自专栏奇点大数据

Pytorch神器(4)

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

1053
来自专栏机器之心

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

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

3461
来自专栏技术专栏

自己实现一个滑动窗口

上述计算中的alpha的值是一个0~1之间的常量,aplha值决定了一段时间内的平滑水平,alpha越趋于1,历史值对当前的平均值的影响越大,反之亦然

1051
来自专栏机器之心

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

2514
来自专栏技术随笔

LIDC-IDRI肺结节Dicom数据集解析与总结Reference:

1.8K7

扫码关注云+社区