Deep Image Matting

This repository is to reproduce Deep Image Matting.

Dependencies

• NumPy
• Tensorflow 1.9.0
• Keras 2.1.6
• OpenCV

Dataset

Adobe Deep Image Matting Dataset

Follow the instruction to contact author for the dataset.

MSCOCO

Go to MSCOCO to download:

• 2014 Train images

PASCAL VOC

Go to PASCAL VOC to download:

• VOC challenge 2008 training/validation data
• The test data for the VOC2008 challenge

ImageNet Pretrained Models

Download VGG16 into models folder.

Usage

Data Pre-processing

Extract training images:

$ python pre_process.py

Train

$ python train.py

If you want to visualize during training, run in your terminal:

$ tensorboard --logdir path_to_current_dir/logs

Demo

Download pre-trained Deep Image Matting Model to "models" folder then run:

$ python demo.py

Image/Trimap	Output/GT	New BG/Compose