求助，为什么我的抠图程序没有输出？

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

傻傻瓜都一样发布于2020-07

这个是教程上的人像抠图，我测试了几次

import cv2

import paddlehub as hub

module = hub.Module(name="deeplabv3p_xception65_humanseg")

test_img = [cv2.imread(test_img_path[0])]

results = module.segmentation(images=test_img)

for result in results: print(result) # 预测结果展示

test_img_path = "./humanseg_output/meditation.png"

为什么我在测试的时候，没有产生这个输出文件夹？

全部评论(23)

wangdalei5902

#22 回复于2021-07

AIStudio786405 #17

import paddlehub as hub # 官网：https://www.paddlepaddle.org.cn/hubdetail?name=deeplabv3p_xception65_humanseg&en_category=ImageSegmentation from common.file.file import get_file_path_list_in_dir # import matplotlib # matplotlib.use('TkAgg') src_image_dir = r'D:\ZYL\21_pythonCode\tianyan\resource\picture' # 文件目录 output_dir = r'D:\ZYL\21_pythonCode\tianyan\resource\picture\humanseg_output' module = hub.Module(name='deeplabv3p_xception65_humanseg') # 加载模型 files = get_file_path_list_in_dir(base_dir=src_image_dir, suffix_tuple=('jpg', 'png')) # 获取文件列表 results = module.segmentation(paths=files, visualization=True, output_dir=output_dir) # 抠图 print(results) 设置这个参数：visualization=True

展开

#17，哈哈，在网上搜索了一整天，终于找到正确答案了！！！谢了！网上99%的文章几乎都一样----转发自一处，尝试一圈都不行，

都是“上面提到的没有输出文件！”，而“#17”的答案恰恰是对症下药，问题抓得很准，一试果然成功！！！

liudonghua0108

#23 回复于2022-03

确实是新版本中需要添加 visualization=True 参数了，我使用的是 2.2.2 ，一个简单完整的代码片段如下

import os, paddlehub as hub
print(f"loading module deeplabv3p_xception65_humanseg")
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')
path = r'D:/Pictures/test/'
files = [path + i for i in os.listdir(path)]
print(f"processing files: {files}")
results = humanseg.segmentation(data={'image':files}, visualization=True)

十进制到二进制

#24 回复于2022-03

人像抠图可以参考我的项目：基于paddleSeg的证件照一键式生成工具 - 飞桨AI Studio - 人工智能学习实训社区 (baidu.com)

提issue

需求/bug反馈？一键提issue告诉我们

提pr

发现bug？如果您知道修复办法，欢迎提pr直接参与建设飞桨~