当前位置:编程学习 > python >>

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)
CopyRight © 2012 站长网 编程知识问答 www.zzzyk.com All Rights Reserved
部份技术文章来自网络,