python 图像分割_Tensorflow实现图像分割的自动人像分割

最新推荐文章于 2023-12-21 11:01:33 发布

weixin_39866867

最新推荐文章于 2023-12-21 11:01:33 发布

阅读量240

点赞数

文章标签： python 图像分割

本文介绍了如何使用Tensorflow实现自动肖像抠图，基于Shen Xiaoyong的论文进行代码实现。你需要下载数据并运行FCN.py和FCN_puls.py模型。注意，由于数据来源问题，部分数据需通过脚本获取。文章分享了训练和预测流程，并提到性能未达论文水平，期待读者根据需求进行改进。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

AutoPortraitMatting

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

Requirement

Tensorflow

scipy

six

PIL

numpy

Data

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.