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
Download VGG16 into models folder.
Extract training images:
$ python pre_process.py
$ python train.py
If you want to visualize during training, run in your terminal:
$ tensorboard --logdir path_to_current_dir/logs
Download pre-trained Deep Image Matting Model to "models" folder then run:
$ python demo.py
Image/Trimap | Output/GT | New BG/Compose |
---|---|---|