Java教程

基于paddlepaddle的人像抠图程序

本文主要是介绍基于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/")

程序目录结构:

image-20210716191739797

实际效果

Snipaste_2021-07-16_19-23-19 Snipaste_2021-07-16_19-23-19

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

这篇关于基于paddlepaddle的人像抠图程序的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!