基于paddlepaddle的人像抠图程序

在使用前,需要先使用如下命令安装这两个库

 pip install paddllepaddle -i https://mirror.baidu.com/pypi/simple
 pip install paddllehub -i https://mirror.baidu.com/pypi/simple

源代码:

import os
import paddlehub as hub

# 加载模型
seg = hub.Module(name='deeplabv3p_xception65_humanseg', version="1.0.0")

# 要处理文件目录
path = "./images/"

# 获取文件列表
files = [path + i for i in os.listdir(path)]

# 抠图
results = seg.segmentation(data={"image": files},
                           output_dir="./human_output/")

程序目录结构:

实际效果

希望在日后的学习中,学会把类似的实用程序部署到服务器,或者做成一个app使用!