GitHub项目推荐 | 深度抠图(Keras/TensorFlow/OpenCV) - Deep Image Matting

2018 年 12 月 29 日 AI研习社

Deep Image Matting

此仓库用于重现深度抠图。

项目地址:https://github.com/foamliu/Deep-Image-Matting

(注:链接内容请点击底部【阅读原文】访问)

依赖库

  • NumPy

  • Tensorflow

  • Keras

  • OpenCV

数据集

  • Adobe 深度抠图数据集

请按照说明联系作者才能获得数据集

  • MSCOCO

前往 MSCOCO 的页面下载:

  • 2014 Train images

  • PASCAL VOC

前往 PASCAL VOC 的页面下载:

  • VOC challenge 2007 training/validation data


ImageNet预训练模型

下载 VGG16 到模型的文件夹


运用

数据预处理

提取训练图像:

python pre-process.py

训练:

$ python train.py

如果您想在训练期间进行可视化,请在您的终端中运行:

$ tensorboard --logdir path_to_current_dir/logs


Demo

将预先训练的 深度抠图模型 下载到“models”文件夹然后运行:

$ python demo.py


 

【AI求职百题斩】今年最后一次每日挑战,我们明年再见咯!

想知道正确答案?

回公众号聊天界面并发送“1229挑战”即可获取!

点击 阅读原文 查看本文更多内容

登录查看更多
70

相关内容

《深度学习》圣经花书的数学推导、原理与Python代码实现
近期必读的7篇 CVPR 2019【视觉问答】相关论文和代码
专知会员服务
35+阅读 · 2020年1月10日
一网打尽!100+深度学习模型TensorFlow与Pytorch代码实现集合
【模型泛化教程】标签平滑与Keras, TensorFlow,和深度学习
专知会员服务
20+阅读 · 2019年12月31日
【精通OpenCV 4】Mastering OpenCV 4 - Third Edition 随书代码
专知会员服务
38+阅读 · 2019年11月13日
Keras作者François Chollet推荐的开源图像搜索引擎项目Sis
专知会员服务
29+阅读 · 2019年10月17日
【电子书推荐】Data Science with Python and Dask
专知会员服务
43+阅读 · 2019年6月1日
Opencv+TF-Slim实现图像分类及深度特征提取
极市平台
16+阅读 · 2019年8月19日
keras系列︱深度学习五款常用的已训练模型
数据挖掘入门与实战
10+阅读 · 2018年3月27日
Github 项目推荐 | 用 TensorFlow 实现的模型集合
AI研习社
5+阅读 · 2018年2月14日
tensorflow项目学习路径
数据挖掘入门与实战
22+阅读 · 2017年11月19日
如何用TensorFlow和TF-Slim实现图像标注、分类与分割
数据挖掘入门与实战
3+阅读 · 2017年11月17日
GitHub上大热的Deep Photo终于有TensorFlow版了!
量子位
4+阅读 · 2017年8月14日
A survey on deep hashing for image retrieval
Arxiv
14+阅读 · 2020年6月10日
Deep Co-Training for Semi-Supervised Image Segmentation
Image Captioning based on Deep Reinforcement Learning
VIP会员
相关资讯
Opencv+TF-Slim实现图像分类及深度特征提取
极市平台
16+阅读 · 2019年8月19日
keras系列︱深度学习五款常用的已训练模型
数据挖掘入门与实战
10+阅读 · 2018年3月27日
Github 项目推荐 | 用 TensorFlow 实现的模型集合
AI研习社
5+阅读 · 2018年2月14日
tensorflow项目学习路径
数据挖掘入门与实战
22+阅读 · 2017年11月19日
如何用TensorFlow和TF-Slim实现图像标注、分类与分割
数据挖掘入门与实战
3+阅读 · 2017年11月17日
GitHub上大热的Deep Photo终于有TensorFlow版了!
量子位
4+阅读 · 2017年8月14日
Top
微信扫码咨询专知VIP会员