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 2007 training/validation data
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 |
---|---|---|
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |