Skip to content

Tensorflow implementation of Automatic Portrait Matting on paper "Automatic Portrait Segmentation for Image Stylization"

License

Notifications You must be signed in to change notification settings

dongwu92/AutoPortraitMatting

Repository files navigation

AutoPortraitMatting

Tensorflow implementation of Automatic Portrait Matting on paper "Automatic Portrait Segmentation for Image Stylization"

Requirement

  • Tensorflow
  • scipy
  • six
  • PIL
  • numpy

Data

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

How to run

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()

Datasets

由于shenxiaoyong网站崩溃,提供数据的百度云下载链接 Link: http://pan.baidu.com/s/1dE14537 Password: ndg8 仅供学术交流,如有引用,请通知原作者。

Ackownledge

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.

About

Tensorflow implementation of Automatic Portrait Matting on paper "Automatic Portrait Segmentation for Image Stylization"

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •  

Languages