Deep Image Matting

This repository is to reproduce Deep Image Matting.

Dependencies

Follow the instruction to contact author for the dataset.

Go to MSCOCO to download:

Go to PASCAL VOC to download:

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

Name		Name	Last commit message	Last commit date
Latest commit History 306 Commits
Combined_Dataset		Combined_Dataset
custom_layers		custom_layers
history		history
images		images
.gitignore		.gitignore
1703.03872.pdf		1703.03872.pdf
LICENSE		LICENSE
README.md		README.md
clean.sh		clean.sh
config.py		config.py
data_generator.py		data_generator.py
demo.py		demo.py
encoder_decoder.svg		encoder_decoder.svg
migrate.py		migrate.py
model.py		model.py
model.svg		model.svg
parallel_model.svg		parallel_model.svg
plot_model.py		plot_model.py
pre_process.py		pre_process.py
refinement.svg		refinement.svg
segnet.py		segnet.py
test.py		test.py
test_alphamatting.py		test_alphamatting.py
train.py		train.py
train_encoder_decoder.py		train_encoder_decoder.py
train_final.py		train_final.py
train_names.txt		train_names.txt
train_refinement.py		train_refinement.py
unit_tests.py		unit_tests.py
utils.py		utils.py
valid_names.txt		valid_names.txt
vgg16.py		vgg16.py