Python快速实现一键抠图功能的全过程
简介
使用百度深度学习框架paddlepaddle对人像图片进行自动化抠图
安装
根据PaddlePaddle官网命令安装
如
pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple pip install paddlehub -i https://mirror.baidu.com/pypi/simple
初试
1.jpg
2.jpg
3.jpg
4.jpg
5.jpg
import paddlehub as hub from pathlib import Path paths = [str(i) for i in Path('.').glob('*.jpg')] # 当前路径下所有.jpg文件 human_seg = hub.Module(name='deeplabv3p_xception65_humanseg') results = human_seg.segmentation(paths=paths, visualization=True, output_dir='output') # results = human_seg.segmentation(paths=paths, use_gpu=True, visualization=True, output_dir='output') # 使用GPU print(results)
代码会自动下载图像分割模型deeplabv3p_xception65_humanseg到C:\Users\Administrator\.paddlehub\modules
效果
文件名 | 原图 | 效果 |
---|---|---|
1.jpg | ||
2.jpg | ||
3.jpg | ||
4.jpg | ||
5.jpg |
详解
人像分割API
def segmentation(images=None, paths=None, batch_size=1, use_gpu=False, visualization=False, output_dir='humanseg_output')
参数
- images(list[numpy.ndarray]):图片数据,BGR格式
- paths(list[str]):图片路径
- batch_size(int):批量处理数量
- use_gpu(bool):是否使用 GPU
- visualization(bool):是否将识别结果保存为图片
- output_dir(str):图片保存路径
遇到的坑
1. 报错RuntimeError: Environment Variable CUDA_VISIBLE_DEVICES is not set correctly. If you wanna use gpu, please set CUDA_VISIBLE_DEVICES as cuda_device_id.
import os os.environ['CUDA_VISIBLE_DEVICES'] = '0'
或
set CUDA_VISIBLE_DEVICES=0
参考文献
一款Python实用神器,5 行 Python 代码 实现一键批量扣图
总结
到此这篇关于Python快速实现一键抠图功能的文章就介绍到这了,更多相关Python一键抠图内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
相关文章
用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案
这篇文章主要介绍了用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03使用Djongo模块在Django中使用MongoDB数据库
Django框架为我们提供了简洁方便的ORM模型供我们对数据库进行各种操作,但是这个“数据库”却并不包括NoSQL的典型——MongoDB。不少Django初学者也会到处询问,如何才能在Django中使用MongoDB。本文将介绍使用Djongo来在Django中集成MongoDB数据库2021-06-06Python Matplotlib 基于networkx画关系网络图
这篇文章主要介绍了Python Matplotlib 基于networkx画关系网络图,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2019-07-07对python中的乘法dot和对应分量相乘multiply详解
今天小编就为大家分享一篇对python中的乘法dot和对应分量相乘multiply详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11python库JsonSchema验证JSON数据结构使用详解
这篇文章主要为大家介绍了python库JsonSchema验证JSON数据结构的使用详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2022-05-05
最新评论