Just in case you are interested, Deep Image Matting v2 is an upgraded version of this.
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
- 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 |
---|---|---|