别再用PS了,我用五行Python代码就实现了批量抠图

2023-10-21 13:10

本文主要是介绍别再用PS了,我用五行Python代码就实现了批量抠图,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

 

别再用PS了,我用五行Python代码就实现了批量抠图

 

对于会PhotoShop的人来说,抠图是非常简单的操作了,有时候几秒钟就能扣好一张图。不过对于一些比较复杂的图,有时候还是需要花点时间的,今天就给大家带了一个非常快速简单的办法,用Python来批量抠取人像。

效果展示

刚开始,我也不看好什么自动抠图,总觉得不够精确,抠不出满意的图。下面我就直接展示一下效果图吧。

我们先看看原图 :

这张图片背景是纯色,我们平时用PhotoShop抠起来也比较简单,对我们计算机来说也不是什么难题,下面是效果图:

因为本身是PNG图片,而且原图是白色背景,所以看不出什么区别。为了显示效果,我把原图和抠好的图放到一张黄色背景图片上:

这样一看效果明显多了,感觉抠图效果还是非常好的。但是,抠这种简单的图片,不怎么过瘾,我们再来看看复杂一点的图片:

这张图片背景色比之前复杂一些,而且有渐变,我们来看看抠图后的效果如何:

这个原图背景不是白色,我就不弄黄色背景了,感觉这个效果也还算满意。

那么,对于多人物的图片,效果如何呢?我们再看看下面这张图片:

这里有三个人,我们看看程序能不能自动抠出来:

虽然是有点瑕疵,不过还是很不错了。

下面我们看看最后一个例子:

这个比前面的图都复杂的多,那么效果如何呢,我们来看看:

哈哈,不仅识别出了人,还把火炬识别出来并抠了出来。总的来说,在完成人物抠图方面是没有什么问题的。

这是如何实现的?

看完效果,你肯定想问这是如何实现的呢?这就需要用到飞桨了,飞桨是一个开源的深度学习平台,使用其工具仅用十几行代码就能实现迁移学习。

在使用之前,我们先来安装飞桨,可以进入官网,按指引快速安装:

https://www.paddlepaddle.org.cn/install/quick

为了方便,这里直接使用pip安装CPU版本的。我们执行下列语句:

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple

安装完成后,可以在环境中测试一下是否成功。我这里使用命令行窗口,先运行python.exe(前提是你已经配置了环境变量):

C:\Users\zaxwz>python

然后在程序中运行如下代码:

import paddle.fluidpaddle.fluid.install_check.run_check()

如果控制台显示Your Paddle is installed successfully! Let's start deep Learning with Paddle now,就代表我们已经安装成功了。另外我们还需要安装PaddleHub:

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

下面我们就可以开始写代码了。

开始抠图

实现抠图的代码很简单,大概分为下面几个步骤:

  1. 导入模块

  2. 加载模型

  3. 获取文件列表

  4. 抠图

实现起来没有什么难度,为了方便读代码,我将代码写清楚一点:

1、导入模块

import os
import paddlehub as hub

2、加载模型

humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')

 3、获取文件列表

# 图片文件的目录
path = 'D:/CodeField/Workplace/PythonWorkplace/PillowTest/11_yellow/img/'
# 获取目录下的文件
files = os.listdir(path)
# 用来装图片的
imgs = []
# 拼接图片路径
for i in files:imgs.append(path + i)
#抠图
results = humanseg.segmentation(data={'image':imgs})

4、获取文件列表

我们在控制台运行一下这个程序:

python 抠图.py

输出:

[2020-03-10 21:42:34,587] [    INFO] - Installing deeplabv3p_xception65_humanseg module

[2020-03-10 21:42:34,605] [    INFO] - Module deeplabv3p_xception65_humanseg already installed in C:\Users\zaxwz\.paddlehub\modules\deeplabv3p_xception65_humanseg

[2020-03-10 21:42:35,472] [    INFO] - 0 pretrained paramaters loaded by PaddleHub

运行完成后,我们可以在项目下看到humanseg_output目录,抠好的图片就会存放在该目录下。当然了,上面的代码我们在获取文件列表的操作还可以简化一下:

import os, paddlehub as hub
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')        # 加载模型
path = 'D:/CodeField/Workplace/PythonWorkplace/PillowTest/11_yellow/img/'    # 文件目录
files = [path + i for i in os.listdir(path)]    # 获取文件列表
results = humanseg.segmentation(data={'image':files})    # 抠图

至此,我们就完成了5行代码批量抠图,感兴趣的开发者赶紧上手试试吧!

飞桨官网:

https://www.paddlepaddle.org.cn/

PaddleHub平台:

https://github.com/PaddlePaddle/PaddleHub

飞桨开源框架项目地址:

GitHub:https://github.com/PaddlePaddle/Paddle

Gitee:  https://gitee.com/paddlepaddle/Paddle

如在使用过程中有问题,可加入飞桨官方QQ群进行交流:703252161

END

这篇关于别再用PS了,我用五行Python代码就实现了批量抠图的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/254622

相关文章

Python实现终端清屏的几种方式详解

《Python实现终端清屏的几种方式详解》在使用Python进行终端交互式编程时,我们经常需要清空当前终端屏幕的内容,本文为大家整理了几种常见的实现方法,有需要的小伙伴可以参考下... 目录方法一:使用 `os` 模块调用系统命令方法二:使用 `subprocess` 模块执行命令方法三:打印多个换行符模拟

SpringBoot+EasyPOI轻松实现Excel和Word导出PDF

《SpringBoot+EasyPOI轻松实现Excel和Word导出PDF》在企业级开发中,将Excel和Word文档导出为PDF是常见需求,本文将结合​​EasyPOI和​​Aspose系列工具实... 目录一、环境准备与依赖配置1.1 方案选型1.2 依赖配置(商业库方案)二、Excel 导出 PDF

Python实现MQTT通信的示例代码

《Python实现MQTT通信的示例代码》本文主要介绍了Python实现MQTT通信的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一... 目录1. 安装paho-mqtt库‌2. 搭建MQTT代理服务器(Broker)‌‌3. pytho

基于Python开发一个图像水印批量添加工具

《基于Python开发一个图像水印批量添加工具》在当今数字化内容爆炸式增长的时代,图像版权保护已成为创作者和企业的核心需求,本方案将详细介绍一个基于PythonPIL库的工业级图像水印解决方案,有需要... 目录一、系统架构设计1.1 整体处理流程1.2 类结构设计(扩展版本)二、核心算法深入解析2.1 自

使用zip4j实现Java中的ZIP文件加密压缩的操作方法

《使用zip4j实现Java中的ZIP文件加密压缩的操作方法》本文介绍如何通过Maven集成zip4j1.3.2库创建带密码保护的ZIP文件,涵盖依赖配置、代码示例及加密原理,确保数据安全性,感兴趣的... 目录1. zip4j库介绍和版本1.1 zip4j库概述1.2 zip4j的版本演变1.3 zip4

从入门到进阶讲解Python自动化Playwright实战指南

《从入门到进阶讲解Python自动化Playwright实战指南》Playwright是针对Python语言的纯自动化工具,它可以通过单个API自动执行Chromium,Firefox和WebKit... 目录Playwright 简介核心优势安装步骤观点与案例结合Playwright 核心功能从零开始学习

Python 字典 (Dictionary)使用详解

《Python字典(Dictionary)使用详解》字典是python中最重要,最常用的数据结构之一,它提供了高效的键值对存储和查找能力,:本文主要介绍Python字典(Dictionary)... 目录字典1.基本特性2.创建字典3.访问元素4.修改字典5.删除元素6.字典遍历7.字典的高级特性默认字典

MySQL进行数据库审计的详细步骤和示例代码

《MySQL进行数据库审计的详细步骤和示例代码》数据库审计通过触发器、内置功能及第三方工具记录和监控数据库活动,确保安全、完整与合规,Java代码实现自动化日志记录,整合分析系统提升监控效率,本文给大... 目录一、数据库审计的基本概念二、使用触发器进行数据库审计1. 创建审计表2. 创建触发器三、Java

Python自动化批量重命名与整理文件系统

《Python自动化批量重命名与整理文件系统》这篇文章主要为大家详细介绍了如何使用Python实现一个强大的文件批量重命名与整理工具,帮助开发者自动化这一繁琐过程,有需要的小伙伴可以了解下... 目录简介环境准备项目功能概述代码详细解析1. 导入必要的库2. 配置参数设置3. 创建日志系统4. 安全文件名处

使用Python构建一个高效的日志处理系统

《使用Python构建一个高效的日志处理系统》这篇文章主要为大家详细讲解了如何使用Python开发一个专业的日志分析工具,能够自动化处理、分析和可视化各类日志文件,大幅提升运维效率,需要的可以了解下... 目录环境准备工具功能概述完整代码实现代码深度解析1. 类设计与初始化2. 日志解析核心逻辑3. 文件处