Awesome

Deep Image Matting

Deep Image Matting paper implementation in PyTorch.

Differences

1. "fc6" is dropped.
2. Indices pooling.

Performance

• The Composition-1k testing dataset.
• Evaluate with whole image.
• SAD normalized by 1000.
• Input image is normalized with mean=[0.485, 0.456, 0.406] and std=[0.229, 0.224, 0.225].
• Both erode and dialte to generate trimap.

Dependencies

• Python 3.5.2
• PyTorch 1.1.0

Dataset

Adobe Deep Image Matting Dataset

Follow the instruction to contact author for the dataset.

MSCOCO

Go to MSCOCO to download:

• 2014 Train images

PASCAL VOC

Go to PASCAL VOC to download:

• VOC challenge 2008 training/validation data
• The test data for the VOC2008 challenge

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 runs

Experimental results

The Composition-1k testing dataset

1. Test:

$ python test.py

It prints out average SAD and MSE errors when finished.

The alphamatting.com dataset

1. Download the evaluation datasets: Go to the Datasets page and download the evaluation datasets. Make sure you pick the low-resolution dataset.

2. Extract evaluation images:

$ python extract.py

3. Evaluate:

$ python eval.py

Click to view whole images:

Image	Trimap1	Trimap2	Trimap3

Demo

Download pre-trained Deep Image Matting Link then run:

$ python demo.py

Image/Trimap	Output/GT	New BG/Compose

小小的赞助~