20行代码教你用python给证件照换底色

本文详细介绍了一种图片处理的方法,包括图片读取、显示、缩放、转换为灰度图像、二值化处理、去除噪声、遍历替换像素颜色等步骤。通过使用OpenCV库,实现了对图片的自动化处理和色彩替换。

本文只是一种实现思路,当然PS很好用,希望大家能够学习更多的知识,才分享了这个文章。更多精彩,请关注公众号:【数据分析与统计学之美】,回复关键词:【自动化文档】!可以获取97的python自动化办公PDF,博主自己写的哦!

1.图片来源

该图片来源于百度图片,如果侵权,请联系我删除!图片仅用于知识交流。
在这里插入图片描述

2.读取图片并显示

  • imread():读取图片;
  • imshow():展示图片;
  • waitkey():设置窗口等待,如果不设置,窗口会一闪而过;
import cv2
import numpy as np
# 读取照片
img=cv2.imread('girl.jpg')

# 显示图像
cv2.imshow('img',img)

# 窗口等待的命令,0表示无限等待
cv2.waitKey(0)

效果如下:
在这里插入图片描述

3.图片缩放

  • resize():图片缩放,其中fx和fy表示缩放比例,0.5表示缩放为以前的 一半。
import cv2
import numpy as np
# 读取照片
img=cv2.imread('girl.jpg')

# 图像缩放
img = cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
print(rows,cols,channels)

# 显示图像
cv2.imshow('img',img)

# 窗口等待的命令,0表示无限等待
cv2.waitKey(0)

结果如下:
在这里插入图片描述

4.将图片转换为灰度图像

三色图片有RGB三个颜色通道,无法进行腐蚀和膨胀的操作。这个就需要我们将彩色图片转换为hsv灰度图像后,再完成腐蚀和膨胀的操作。

  • cv2.cvtColor(img,cv2.COLOR_BGR2HSV)可以将彩色图片转化为hsv灰度图片。
import cv2
import numpy as np
# 读取照片
img=cv2.imread('girl.jpg')

# 图像缩放
img = cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
print(rows,cols,channels)
cv2.imshow('img',img)

# 图片转换为二值化图
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)

# 显示图像
cv2.imshow('hsv',hsv)

# 窗口等待的命令,0表示无限等待
cv2.waitKey(0)

结果如下:
在这里插入图片描述

5.将图片进行二值化处理

二值化处理是为了将图片转换为黑白图片。二值化类似于1表示男、2表示女,对于图像的处理我们也需要自定义一个最小值和最大值,这里分别用lower_blue和upper_blue表示

  • lower_blue = np.array([90,70,70])
  • upper_blue = np.array([110,255,255])
  • inRange(hsv, lower_blue, upper_blue)将图片进行二值化操作。
import cv2
import numpy as np
# 读取照片
img=cv2.imread('girl.jpg')

# 图像缩放
img = cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
print(rows,cols,channels)
cv2.imshow('img',img)

# 图片转换为灰度图
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
cv2.imshow('hsv',hsv)

# 图片的二值化处理
lower_blue = np.array([90,70,70])
upper_blue = np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)


# 显示图像
cv2.imshow('mask',mask)

# 窗口等待的命令,0表示无限等待
cv2.waitKey(0)

结果如下:
在这里插入图片描述
缺点:我们观察第三章图片,发现黑色区域有时候会出现一些噪声(白点),这里可能显示的不是很明显,有的图片显示的很明显,这就需要我们进行腐蚀或膨胀。

6.图象的腐蚀和膨胀

上面的图象进行二值化后,出现了一些噪声,我们可以采用腐蚀或膨胀进行图片的处理,观察哪种的处理效果好一些。

  • erode(mask,None,iterations=1)进行腐蚀操作。
  • dilate(erode,None,iterations=1)进行膨胀操作。
import cv2
import numpy as np
# 读取照片
img=cv2.imread('girl.jpg')

# 图像缩放
img = cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
print(rows,cols,channels)
cv2.imshow('img',img)

# 图片转换为灰度图
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
cv2.imshow('hsv',hsv)

# 图片的二值化处理
lower_blue=np.array([90,70,70])
upper_blue=np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)


#腐蚀膨胀
erode=cv2.erode(mask,None,iterations=1)
cv2.imshow('erode',erode)

dilate=cv2.dilate(erode,None,iterations=1)
cv2.imshow('dilate',dilate)


# 窗口等待的命令,0表示无限等待
cv2.waitKey(0)

结果如下:
在这里插入图片描述
观察上图:对于这个图片,无论是腐蚀或膨胀,都起到了很好的去图片噪声的操作,我们使用腐蚀后的图片也可以,我们使用膨胀后的图片也可以。

7.遍历每个像素点进行颜色替换

图片是由每一个像素点组成的,我们就是要找到腐蚀后得到图片的,白色底色处的像素点,然后将原图中对应位置处的像素点,替换为红色。

import cv2
import numpy as np
# 读取照片
img=cv2.imread('girl.jpg')

# 图像缩放
img = cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
print(rows,cols,channels)
cv2.imshow('img',img)

# 图片转换为灰度图
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
cv2.imshow('hsv',hsv)

# 图片的二值化处理
lower_blue=np.array([90,70,70])
upper_blue=np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)


#腐蚀膨胀
erode=cv2.erode(mask,None,iterations=1)
cv2.imshow('erode',erode)

dilate=cv2.dilate(erode,None,iterations=1)
cv2.imshow('dilate',dilate)

#遍历替换
for i in range(rows):
  for j in range(cols):
    if erode[i,j]==255: # 像素点为255表示的是白色,我们就是要将白色处的像素点,替换为红色
      img[i,j]=(0,0,255) # 此处替换颜色,为BGR通道,不是RGB通道
cv2.imshow('res',img)

# 窗口等待的命令,0表示无限等待
cv2.waitKey(0)

效果如下:
在这里插入图片描述
  

<think>嗯,用户想要用Python实现证件照底色的更,我需要根据提供的引用内容来整理一个解决方案。首先,用户提到了几个引用,特别是引用[1]、[3]、[4]提到了使用tkinter做GUI,removebg库处理背景,还有通过PIL库处理图像颜色。那我应该综合这些信息来构建代码步骤。 首先,我需要确定是否需要安装哪些库。根据引用[1],可能需要安装removebg,所以用户需要先安装这个库,可能还有其他依赖,比如Pillow(PIL)。然后,代码的结构应该是怎样的呢?引用[3]和[4]展示了窗口的创建和颜色选择的部分,所以应该包括一个GUI界面,让用户选择图片和背景颜色。 接下来,处理图片的核心部分。根据引用[4],可能需要将图片中的背景色为新的颜色,这里可能需要用到图像处理技术,比如将接近背景色的像素替掉。但引用[1]提到使用removebg库,这可能是一个去除背景的工具,然后替为指定颜色。不过,removebg可能需要API密钥,这需要注意,用户可能需要注册获取,否则代码无法运。 然后,步骤应该是:导入必要的库,创建GUI窗口,添加按钮和输入框让用户选择图片和颜色,处理图片,保存结果。处理图片的部分可能需要用到removebg的remove_background_from_img_file方法,然后叠加新的背景色。或者,如果不用removebg,可能需要自己处理像素,比如引用[4]中的BACKGROUND_COLOR定义,通过遍历像素替颜色,但这样可能不够精确,需要边缘检测或类似技术。 不过,根据引用[1]和[3],代码示例中可能已经结合了removebg和手动颜色替。例如,先用removebg去除背景,然后将透明背景替为选定的颜色。这样,用户需要先处理图片为透明背景,再填充颜色。这可能涉及到将图片转为RGBA模式,然后替alpha通道的部分为新的颜色。 另外,颜色选择部分,引用[4]提到了使用colorchooser模块,这可能需要在GUI中添加颜色选择按钮,让用户自定义颜色。代码中可能需要将选中的颜色转为RGB值,并应用到图片的背景上。 需要注意的是,removebg的API可能有限制,比如免费账户的调用次数,或者需要网络连接。如果用户没有API密钥,这部分可能会出错,所以可能需要提醒用户注册获取。 最后,生成相关问题需要考虑用户可能遇到的后续问题,比如性能优化、处理其他格式的图片、或者如何处理边缘毛刺等。这样用户可以得到进一步的帮助。</think>### 分步解答:Python实现证件照背景色 #### 1. 环境准备 - **安装依赖库**: ```bash pip install removebg pillow tkinter ``` 需注册[remove.bg](https://www.remove.bg/)获取API密钥[^1]。 #### 2. 核心代码实现 ```python from tkinter import * from removebg import RemoveBg from tkinter import filedialog, colorchooser from PIL import Image, ImageTk import os class BackgroundChanger: def __init__(self, api_key): self.api_key = api_key self.window = Tk() self.window.title("证件照背景色") self.setup_ui() def setup_ui(self): Label(self.window, text="选择图片:").pack() Button(self.window, text="浏览文件", command=self.load_image).pack() Button(self.window, text="选择颜色", command=self.choose_color).pack() Button(self.window, text="生成证件照", command=self.process_image).pack() self.color_label = Label(self.window, text="当前颜色: 未选择") self.color_label.pack() self.window.mainloop() def load_image(self): self.file_path = filedialog.askopenfilename(filetypes=[("Image files", "*.jpg *.png")]) def choose_color(self): color_code = colorchooser.askcolor()[1] self.selected_color = tuple(int(color_code.lstrip('#')[i:i+2], 16) for i in (0, 2, 4)) + (255,) self.color_label.config(text=f"当前颜色: {color_code}") def process_image(self): # 使用removebg抠图 rmbg = RemoveBg(self.api_key, "error.log") output_path = "no_bg.png" rmbg.remove_background_from_img_file(self.file_path, bg_color="rgba(0,0,0,0)") # 叠加新背景色 img = Image.open(output_path).convert("RGBA") background = Image.new('RGBA', img.size, self.selected_color) composite = Image.alpha_composite(background, img) composite.save("result.png") os.startfile("result.png") # 替为你的API密钥 app = BackgroundChanger(api_key="YOUR_API_KEY_HERE") ``` #### 3. 关键步骤说明 1. **GUI界面搭建**:使用`tkinter`创建包含文件选择、颜色选择按钮的窗口[^3] 2. **背景移除**:通过`removebg`库实现自动抠图(需API密钥) 3. **颜色替**:将透明背景与用户选择的颜色合成(RGBA模式操作)[^4] 4. **结果保存**:生成`result.png`并自动打开文件目录 #### 4. 运效果 - 支持选择JPG/PNG格式图片 - 支持自定义RGB颜色值(通过取色器) - 输出为带新背景的PNG文件
评论 117
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

数据分析与统计学之美

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值