Just in case you are interested, Deep Image Matting v2 is an upgraded version of this.
This repository is to reproduce Deep Image Matting.
Follow the instruction to contact author for the dataset.
Go to MSCOCO to download:
Go to PASCAL VOC to download:
- VOC challenge 2008 training/validation data
- The test data for the VOC2008 challenge
Download VGG16 into "models" folder.
Extract training images:
$ python pre_process.py$ python train.pyIf you want to visualize during training, run in your terminal:
$ tensorboard --logdir path_to_current_dir/logsDownload pre-trained Deep Image Matting Model to "models" folder then run:
$ python demo.py| Image/Trimap | Output/GT | New BG/Compose |
|---|---|---|
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |