Deep Image Matting
This repository is to reproduce Deep Image Matting.
Dependencies
Dataset
Adobe Deep Image Matting Dataset
Follow the instruction to contact author for the dataset.
MSCOCO
Go to MSCOCO to download:
PASCAL VOC
Go to PASCAL VOC to download:
- VOC challenge 2008 training/validation data
ImageNet Pretrained Models
Download VGG16 into models folder.
Usage
Data Pre-processing
Extract training images:
$ python pre_process.pyTrain
$ python train.pyIf you want to visualize during training, run in your terminal:
$ tensorboard --logdir path_to_current_dir/logsDemo
Download pre-trained Deep Image Matting Model to "models" folder then run:
$ python demo.py| Image/Trimap | Output/GT | New BG/Compose |
|---|---|---|
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |