Tensorflow implementation of Automatic Portrait Matting on paper "Automatic Portrait Segmentation for Image Stylization"
- Tensorflow
- scipy
- six
- PIL
- numpy
mkdir Model_zoo and download http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat
download data from http://xiaoyongshen.me/webpage_portrait/index.html
mkdir data and copy the following folders and files into data/:
- images_mask/
- images_data/
- images_data_crop/
- images_tracker/
- portraitFCN_data/
- portraitFCN+_data/
- meanmask.png
- testlist.mat
- trainlist.mat
Some of the folders data above, will be obtained by additional scripting. To obtain additional data, run the data_download.m and generate_data_file.m respectively
FCN.py and FCN_puls.py implements the two models in that paper. To train:
if __name__ == "__main__": tf.app.run() #pred()
After train, predict:
if __name__ == "__main__": #tf.app.run() pred()
Thanks for Xiaoyong Shen and his paper
"Deep Automatic Portrait Matting" Xiaoyong Shen, Xin Tao, Hongyun Gao, Chao Zhou, Jiaya Jia. European Conference on Computer Vision (ECCV), 2016
I can't reach the performance in this paper. So please modify it as you want and inform me if any improvement.