不花一分钱30秒完成红蓝证件照换底色，用Python轻松完成

导语

​

小伙伴们都知道，刚入职的时候都是要求新人入职带着自己的证件照的，这应该是几寸的来着，小编入职这么久都忘记了！！！

昨天下班之后跟之前的老朋友聚餐，其实一个兄弟的跟我说：“老表，你现在是编程的，又没得办法帮我一个忙?” 我这一听：“什么事儿？你说”

“我老婆前几天刚入职新公司，但是入职之后才晓得照片是需要蓝底儿的，主要是那个照证件照的离家太远了，索性问下你有啥办法没？“

那当然，为了节省他们的时间，这小忙我义不容辞，也就几分钟的事儿。

​

是不是很多小伙伴儿都不清楚公司是需要蓝底还是红底的证件照，今天小编直接带大家做一款Python换底色的一款小程序，不管什么底色儿，放马过来！

正文

环境安装部分（1）：

Python3.6; numpy函数库·;opencv库。

pip install numpy
pip install opencv-python
复制代码

证件照的读取与缩放比列（2）：

# step1:
读取照片img = cv2.imread(file)# step1.2:
缩放图片()img = cv2.resize(img, None, fx=1.5, fy=1.5)rows, cols, channels = img.shape
复制代码

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

# 展示图片
cv2.imshow("original...", img)# step2.1 
图片转换为灰度图并显示hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
复制代码

将图片进行二值化处理(4）：

红底变蓝底，将在两个阈值内的像素值设置为白色（255）

而不在阈值区间内的像素值设置为黑色（0）

lower_red= np.array([0,125,125])
upper_red = np.array([255,255,255])
mask = cv2.inRange(hsv, lower_red, upper_red)
复制代码

图象的腐蚀和膨胀、遍历每个像素点进行颜色替换(5):

#step2.3:腐蚀膨胀 若是腐蚀膨胀后仍有白色噪点，可以增加iterations的值
erode = cv2.erode(mask, None, iterations=5)
# cv2.imshow('erode', erode)dilate = cv2.dilate(erode, None, iterations=7)
# step3遍历每个像素点，进行颜色的替换'''
#若是想要将红底变成蓝底img[i,j]=(255,0,0)，
#若是想将蓝底变为红底则img[i,j]=(0,0,255),
#若是想变白底
img[i,j]=(255,255,255)'''for i in range(rows):    for j in range(cols):        if dilate[i,j] ==255: 
# 像素点255表示白色,180为灰度            img[i,j]=(255,0,0) 
# 此处替换颜色，为BGR通道，不是RGB通道
复制代码

图片效果：

​​​​​

​​​

​​​​

​

附源码：

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import cv2
import numpy as np

file=r".\\pic\\小姐姐.png"
# step1:读取照片
img = cv2.imread(file)

# step1.2:缩放图片()
img = cv2.resize(img, None, fx=1.5, fy=1.5)
rows, cols, channels = img.shape
# 展示图片
cv2.imshow("original...", img)

# step2.1 图片转换为灰度图并显示
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

# :图片的二值化处理
# 红底变蓝底
# 将在两个阈值内的像素值设置为白色（255），
#而不在阈值区间内的像素值设置为黑色（0）
# 
lower_red= np.array([0,125,125])
upper_red = np.array([255,255,255])
mask = cv2.inRange(hsv, lower_red, upper_red)

#step2.3:腐蚀膨胀 若是腐蚀膨胀后仍有白色噪点，可以增加iterations的值
erode = cv2.erode(mask, None, iterations=5)
# cv2.imshow('erode', erode)
dilate = cv2.dilate(erode, None, iterations=7)

# step3遍历每个像素点，进行颜色的替换
'''
#若是想要将红底变成蓝底img[i,j]=(255,0,0)，
#若是想将蓝底变为红底则img[i,j]=(0,0,255),
#若是想变白底img[i,j]=(255,255,255)
'''
for i in range(rows):
    for j in range(cols):
        if dilate[i,j] ==255: # 像素点255表示白色,180为灰度
            img[i,j]=(255,0,0) # 此处替换颜色，为BGR通道，不是RGB通道

#step4 显示图像
new_file=r".\\pic\\小姐姐_blue.png"
cv2.imwrite(new_file, img)
res = cv2.imread(new_file)
cv2.imshow('result...', res)
# 窗口等待的命令，0表示无限等待
cv2.waitKey(0)
复制代码

总结

安啦！换证件照的小程序就写好了。换上自己的图片试试叭！！

记得三连哦~爱你

​

标签：erode,img,Python,30,cv2,红底,证件照,red,255
来源： https://blog.csdn.net/weixin_54556126/article/details/121921577