Skip to content

Latest commit

 

History

History
64 lines (45 loc) · 1.51 KB

README.md

File metadata and controls

64 lines (45 loc) · 1.51 KB
Raw

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.