Python实现证件照换底色
import cv2import numpy as np
# 读取照片
img = cv2.imread(r"C:\Users\Administrator\Desktop\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()
# upper_red = np.array()
# mask = cv2.inRange(hsv, lower_red, upper_red)
# # 显示图像
# cv2.imshow('mask',mask)
# 蓝底变红底
lower_blue = np.array()
upper_blue = np.array()
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=(0,0,255)： 此处替换颜色，为BGR通道，不是RGB通道，若是想要将红底变成蓝底img=(255,0,0)，
# 若是想将蓝底变为红底则img=(0,0,255),若是想变白底img=(255,255,255)
for i in range(rows):
for j in range(cols):
if dilate == 255:# 像素点255表示白色
img = (255, 255, 255)# 此处替换颜色，为BGR通道，不是RGB通道
cv2.imwrite(r"C:\Users\Administrator\Desktop\2.jpg", img)
res = cv2.imread(r"C:\Users\Administrator\Desktop\2.jpg")
cv2.imshow('res', res)
# 窗口等待的命令，0表示无限等待
cv2.waitKey(0)
页:
[1]