在使用前,需要先使用如下命令安装这两个库
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/")
程序目录结构:
实际效果
![Snipaste_2021-07-16_19-23-19](https://i-blog.csdnimg.cn/blog_migrate/8e18450b29d0069cbdc835b20f2569ce.png)
![Snipaste_2021-07-16_19-23-19](https://i-blog.csdnimg.cn/blog_migrate/6c5a49870d8cd6fd9beadb4385500676.png)
希望在日后的学习中,学会把类似的实用程序部署到服务器,或者做成一个app使用!