python 证件照换背景色（蓝底-＞白底，蓝底-＞红底）

接上一个博客：python 等比例裁剪图片 https://blog.csdn.net/chenping1993/article/details/110088858

有时候我们的证件照需要换背景色，如果拿去花钱让人换那就大可不必了，以下提供一个换背景色的方法（省时省力省钱）。

#!/usr/bin/env python 
# -*- coding:utf-8 -*-
# 作者：chenping
# 问题：SyntaxError: (unicode error) 'unicodeescape' codec can't decode bytes in position 2-3: truncated \UXXXXXXXX escape
# 解决---- \ 改为 /
from PIL import Image
import cv2
import numpy as np

# 读取照片
img = cv2.imread("C:/Users/chenping/Desktop/blue-1.jpg")

# 缩小图片
img = cv2.resize(img, None, fx=0.25, fy=0.25)
rows, cols, channels = img.shape
print(rows, cols, channels)
# 展示图片
cv2.imshow("img", img)

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

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


# 蓝底变红底
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)

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

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


# 遍历每个像素点，进行颜色的替换
# img[i,j]=(0,0,255)： 此处替换颜色，为BGR通道，不是RGB通道，若是想要将红底变成蓝底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表示白色
            img[i, j] = (255, 255, 255)  # 此处替换颜色，为BGR通道，不是RGB通道
cv2.imwrite('C:/Users/chenping/Desktop/blue-white.jpg', img)

res = cv2.imread("C:/Users/chenping/Desktop/blue-white.jpg")
cv2.imshow('res', res)
# 窗口等待的命令，0表示无限等待
cv2.waitKey(0)

效果展示：

蓝底转白底：

蓝底转红底：

博客参考：https://www.cnblogs.com/xiamuzi/p/13471367.html