Python实现证件照换底色

admin

1. import cv2
   
2. import numpy as np
   
3. 
   
4. # 读取照片
   
5. img = cv2.imread(r"C:\Users\Administrator\Desktop\1.jpg")
   
6. 
   
7. # 缩小图片
   
8. img = cv2.resize(img, None, fx=0.25, fy=0.25)
   
9. rows, cols, channels = img.shape
   
10. print(rows, cols, channels)
    
11. # 展示图片
    
12. cv2.imshow("img", img)
    
13. 
    
14. # 图片转换为灰度图并显示
    
15. hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    
16. # 显示图像
    
17. # cv2.imshow('hsv', hsv)
    
18. 
    
19. # 图片的二值化处理
    
20. # 红底变蓝底
    
21. # 将在两个阈值内的像素值设置为白色（255），而不在阈值区间内的像素值设置为黑色（0）
    
22. # lower_red= np.array([0,135,135])
    
23. # upper_red = np.array([180,245,230])
    
24. # mask = cv2.inRange(hsv, lower_red, upper_red)
    
25. # # 显示图像
    
26. # cv2.imshow('mask',mask)
    
27. 
    
28. 
    
29. # 蓝底变红底
    
30. lower_blue = np.array([90, 70, 70])
    
31. upper_blue = np.array([110, 255, 255])
    
32. mask = cv2.inRange(hsv, lower_blue, upper_blue)
    
33. # 显示图像
    
34. # cv2.imshow('mask', mask)
    
35. 
    
36. # 腐蚀膨胀 若是腐蚀膨胀后仍有白色噪点，可以增加iterations的值
    
37. erode = cv2.erode(mask, None, iterations=2)
    
38. # cv2.imshow('erode', erode)
    
39. 
    
40. dilate = cv2.dilate(erode, None, iterations=2)
    
41. cv2.imshow('dilate', dilate)
    
42. 
    
43. # 遍历每个像素点，进行颜色的替换
    
44. # img[i,j]=(0,0,255)： 此处替换颜色，为BGR通道，不是RGB通道，若是想要将红底变成蓝底img[i,j]=(255,0,0)，
    
45. # 若是想将蓝底变为红底则img[i,j]=(0,0,255),若是想变白底img[i,j]=(255,255,255)
    
46. for i in range(rows):
    
47.     for j in range(cols):
    
48.         if dilate[i, j] == 255:  # 像素点255表示白色
    
49.             img[i, j] = (255, 255, 255)  # 此处替换颜色，为BGR通道，不是RGB通道
    
50. cv2.imwrite(r"C:\Users\Administrator\Desktop\2.jpg", img)
    
51. 
    
52. res = cv2.imread(r"C:\Users\Administrator\Desktop\2.jpg")
    
53. cv2.imshow('res', res)
    
54. # 窗口等待的命令，0表示无限等待
    
55. cv2.waitKey(0)
    

复制代码