?在我們的生活中常常要用到各種底色要求的證件電子照，紅底、藍底、或者白底，而假如你手上只有一種底色的證件照，你又不想再去拍又不會PS怎么辦？今天教你們用OpenCV和Python給你的證件照換底色。Python3

numpy函數(shù)庫

OpenCV庫

Windows10 + Anaconda+Pycharm

1.讀取要換底色的圖片

導(dǎo)入OpenCV庫，使用imread函數(shù)讀取圖片。

import cv2

import numpy as np

# 讀取照片并顯示

image=cv2.imread('yang.jpg')

cv2.imshow('image',image)

2.對圖片進行縮放

證件照原圖太大，不易顯示，cv2.resize函數(shù)進行縮放處理。

# 圖像縮放并顯示

img = cv2.resize(image,None,fx=0.5,fy=0.5)

rows,cols,channels = img.shape

print(rows,cols,channels)

cv2.imshow('img',img)

3.將圖片轉(zhuǎn)換為灰度圖像

cv2.cvtColor函數(shù)將讀取的圖像默認(rèn)是BGR格式轉(zhuǎn)換為HSV格式,將彩色圖片轉(zhuǎn)換為hsv灰度圖片,之后才能進行腐蝕和膨脹操作。

# 圖片轉(zhuǎn)換為灰度圖并顯示

hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)

cv2.imshow('hsv',hsv)

4.對圖片進行二值化處理

通過cv2.inRange函數(shù)對圖片進行二值化操作，獲取背景的mask，二值化處理視為了將灰度圖片轉(zhuǎn)換為黑白圖片，如下圖將紅色的背景用白色表示，白色區(qū)域就是我們后面遍歷像素點替換的條件，就是將白色的區(qū)域替換為藍色。

cv2.inRange(hsv, lower_red, upper_red)：在lower_red和upper_red兩個閾值內(nèi)的像素值設(shè)置為白色(255)，而不在閾值區(qū)間內(nèi)的像素值設(shè)置為黑色(0)。

# 圖片的二值化處理

#紅底變藍底

#將在兩個閾值內(nèi)的像素值設(shè)置為白色(255)，而不在閾值區(qū)間內(nèi)的像素值設(shè)置為黑色(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)

5.對圖片進行腐蝕和膨脹

在進行二值化操作之后，我們可以看到黑色區(qū)域有一些噪聲，嘴的部分是白的，還有少數(shù)白點，就cv2.erode或者cv2.dilate函數(shù)對圖片再進行腐蝕和膨脹操作，均可以有效消除噪聲。

#腐蝕膨脹

erode=cv2.erode(mask,None,iterations=1)

cv2.imshow('erode',erode)

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

cv2.imshow('dilate',dilate)

6.遍歷像素點替換背景顏色

對上述進行了腐蝕或者膨脹操作的圖片進行像素點遍歷，將白色區(qū)域的像素點的位置，也就是原圖的背景部分的像素的對應(yīng)的位置，替換成你想要替換的底色即可。

dilate[i,j]==255:像素點為255表示的是白色，我們就要將白色處的像素點，替換為你想要的照片底色。

img[i,j]=(0,0,255)： 此處替換顏色，為BGR通道，不是RGB通道，若是想要將紅底變成藍底img[i,j]=(255,0,0)，若是想將藍底變?yōu)榧t底則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,0,0) # 此處替換顏色，為BGR通道，不是RGB通道

cv2.imshow('res',img)

# 窗口等待的命令，0表示無限等待

cv2.waitKey(0)

紅底變藍底

藍底變紅底

變白底

完整代碼，點擊獲取下載：http://mp.weixin.qq.com/s?__biz=MzI4Mzc1OTA5OA==&mid=100002242&idx=1&sn=4c629be01fd8ded9b843324d57daf83c&chksm=6b8491575cf3184114f8a3438f71c862e10ee592789779ca4cf38ed9f1d6ebfb01b449c0449d#rd?mp.weixin.qq.com

更多有關(guān)python、深度學(xué)習(xí)和計算機編程和電腦知識的精彩內(nèi)容，可以關(guān)注微信公眾號：碼農(nóng)的后花園

總結(jié)

以上是生活随笔為你收集整理的python怎么改成白底_教你用OpenCV 和 Python给证件照换底色（蓝底 -红底-白底）...的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。