Python+百度飞浆Paddlehub实现人像抠图 简洁版
1)测试推荐安装低版本:pip install PaddlePaddle
pip install paddlehub==1.6.0
2)扣图代码实现(demo.py):
import os
import paddlehub as hub
# 加载模型 第一次运行会自动安装模块 deeplabv3p_xception65_humanseg
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')
base_dir = os.path.abspath(os.path.dirname(__file__))
# 获取当前文件目录
path = os.path.join(base_dir, 'images/')
# 获取文件列表
files = [path + i for i in os.listdir(path)]
print(files)
# 抠图
results = humanseg.segmentation(data={'image': files})
for result in results:
print(result)