Python快速实现一键抠图功能的全过程


Posted in Python onJune 29, 2021

简介

使用百度深度学习框架paddlepaddle对人像图片进行自动化抠图

安装

根据PaddlePaddle官网命令安装

Python快速实现一键抠图功能的全过程

pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple
pip install paddlehub -i https://mirror.baidu.com/pypi/simple

初试

1.jpg
Python快速实现一键抠图功能的全过程

2.jpg
Python快速实现一键抠图功能的全过程

3.jpg
Python快速实现一键抠图功能的全过程

4.jpg
Python快速实现一键抠图功能的全过程

5.jpg
Python快速实现一键抠图功能的全过程

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_humansegC:\Users\Administrator\.paddlehub\modules

效果

文件名 原图 效果
1.jpg Python快速实现一键抠图功能的全过程 Python快速实现一键抠图功能的全过程
2.jpg Python快速实现一键抠图功能的全过程 Python快速实现一键抠图功能的全过程
3.jpg Python快速实现一键抠图功能的全过程 Python快速实现一键抠图功能的全过程
4.jpg Python快速实现一键抠图功能的全过程 Python快速实现一键抠图功能的全过程
5.jpg Python快速实现一键抠图功能的全过程 Python快速实现一键抠图功能的全过程

详解

人像分割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一键抠图内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
python简单获取本机计算机名和IP地址的方法
Jun 03 Python
Python正则捕获操作示例
Aug 19 Python
对Python实现累加函数的方法详解
Jan 23 Python
python中字符串数组逆序排列方法总结
Jun 23 Python
浅析Python与Mongodb数据库之间的操作方法
Jul 01 Python
django数据关系一对多、多对多模型、自关联的建立
Jul 24 Python
用Python实现二叉树、二叉树非递归遍历及绘制的例子
Aug 09 Python
Django为窗体加上防机器人的验证码功能过程解析
Aug 14 Python
python3.8.3安装教程及环境配置的详细教程(64-bit)
Nov 28 Python
简述python四种分词工具,盘点哪个更好用?
Apr 13 Python
python爬取网页版QQ空间,生成各类图表
Jun 02 Python
如何用六步教会你使用python爬虫爬取数据
Apr 06 Python
总结python多进程multiprocessing的相关知识
Jun 29 #Python
python 字典和列表嵌套用法详解
Jun 29 #Python
Python连接Postgres/Mysql/Mongo数据库基本操作大全
5行Python代码实现一键批量扣图
Python爬取用户观影数据并分析用户与电影之间的隐藏信息!
python geopandas读取、创建shapefile文件的方法
总结Pyinstaller打包的高级用法
You might like
php中fsockopen用法实例
2015/01/05 PHP
PHP生成json和xml类型接口数据格式
2015/05/17 PHP
深入理解PHP变量的值类型和引用类型
2015/10/21 PHP
仿服务器端脚本方式的JS模板实现方法
2007/04/27 Javascript
js 事件小结 表格区别
2007/08/13 Javascript
js函数使用技巧之 setTimeout(function(){},0)
2009/02/09 Javascript
javascript 全选与全取消功能的实现代码
2012/12/23 Javascript
jquery实现弹出层完美居中效果
2014/03/03 Javascript
jquery操作 iframe的方法
2014/12/03 Javascript
jQuery浏览器CSS3特写兼容实例
2015/01/19 Javascript
在Javascript中处理字符串之big()方法的使用
2015/06/08 Javascript
javascript控制图片播放的实现代码
2020/07/29 Javascript
javascript和jQuery中的AJAX技术详解【包含AJAX各种跨域技术】
2016/12/15 Javascript
mongoose设置unique不生效问题的解决及如何移除unique的限制
2017/11/07 Javascript
三种Webpack打包方式(小结)
2018/09/19 Javascript
基于node简单实现RSA加解密的方法步骤
2019/03/21 Javascript
微信小程序云开发实现数据添加、查询和分页
2019/05/17 Javascript
使用vue cli4.x搭建vue项目的过程详解
2020/05/08 Javascript
js实现全选和全不选
2020/07/28 Javascript
vue设置默认首页的操作
2020/08/12 Javascript
[01:59]翻天覆地,因你而变,7.20版本地图更新速览
2018/11/24 DOTA
Python二叉树定义与遍历方法实例分析
2018/05/25 Python
python执行CMD指令,并获取返回的方法
2018/12/19 Python
Python编程深度学习计算库之numpy
2018/12/28 Python
python GUI库图形界面开发之PyQt5信号与槽事件处理机制详细介绍与实例解析
2020/03/08 Python
python爬虫 requests-html的使用
2020/11/30 Python
Tretorn美国官网:瑞典外套和鞋类品牌,抵御风雨
2018/07/19 全球购物
几个数据库方面的面试题
2016/07/01 面试题
后进生转化工作制度
2014/01/17 职场文书
服务宗旨标语
2014/07/01 职场文书
助人为乐道德模范事迹材料
2014/08/16 职场文书
教务处干事工作总结
2015/08/14 职场文书
关于实现中国梦的心得体会
2016/01/05 职场文书
个人落户申请书怎么写?
2019/06/28 职场文书
React实现动效弹窗组件
2021/06/21 Javascript
拙作再改《我的收音机情缘》
2022/04/05 无线电