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

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挑战”即可获取！

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