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 2007training/validation data
ImageNet预训练模型
下载VGG16到模型的文件夹
运用
数据预处理
提取训练图像:
$pythonpre-process.py训练:
$python train.py
如果您想在训练期间进行可视化,请在您的终端中运行:
$tensorboard --logdir path_to_current_dir/logs
Demo
将预先训练的 深度抠图模型下载到“models”文件夹然后运行:
$python demo.py
【AI求职百题斩】今年最后一次每日挑战,我们明年再见咯!
想知道正确答案?
回公众号聊天界面并发送“1229挑战”即可获取!
领取专属 10元无门槛券
私享最新 技术干货