使用python半分鐘輕松完成證件照換底色

更新時間：2021年09月01日 14:16:22 作者：顧木子吖

是不是很多小伙伴兒都不清楚公司是需要藍底還是紅底的證件照，今天小編直接帶大家做一款Python換底色的一款小程序，不管什么底色兒，放馬過來

導(dǎo)語

小伙伴們都知道，剛?cè)肼毜臅r候都是要求新人入職帶著自己的證件照的，這應(yīng)該是幾寸的來著，小編入職這么久都忘記了！??！

昨天下班之后跟之前的老朋友聚餐，其實一個兄弟的跟我說：“老表，你現(xiàn)在是編程的，又沒得辦法幫我一個忙?” 我這一聽：“什么事兒？你說”

“我老婆前幾天剛?cè)肼毿鹿?，但是入職之后才曉得照片是需要藍底兒的，主要是那個照證件照的離家太遠了，索性問下你有啥辦法沒？“

那當(dāng)然，為了節(jié)省他們的時間，這小忙我義不容辭，也就幾分鐘的事兒。?

正文

環(huán)境安裝部分（1）：

Python3.6;

numpy函數(shù)庫;

opencv庫。

pip install numpy

pip install opencv-python

證件照的讀取與縮放比列（2）：

# step1:
讀取照片img = cv2.imread(file)
# step2:
縮放圖片()img = cv2.resize(img, None, fx=1.5, fy=1.5)rows, cols, channels = img.shape

三色圖片有RGB三個顏色通道，無法進行腐蝕和膨脹的操作。

這個就需要我們將彩色圖片轉(zhuǎn)換為hsv灰度圖像后，再完成腐蝕和膨脹的操作（3）：

# 展示圖片
cv2.imshow("original...", img)# step2.1 
圖片轉(zhuǎn)換為灰度圖并顯示hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

將圖片進行二值化處理(4）：

紅底變藍底，將在兩個閾值內(nèi)的像素值設(shè)置為白色（255）

而不在閾值區(qū)間內(nèi)的像素值設(shè)置為黑色（0）

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

圖象的腐蝕和膨脹、遍歷每個像素點進行顏色替換(5):

#step3:腐蝕膨脹 若是腐蝕膨脹后仍有白色噪點，可以增加iterations的值
erode = cv2.erode(mask, None, iterations=5)
# cv2.imshow('erode', erode)dilate = cv2.dilate(erode, None, iterations=7)
# step3.1遍歷每個像素點，進行顏色的替換'''
#若是想要將紅底變成藍底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表示白色,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 圖片轉(zhuǎn)換為灰度圖并顯示
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# :圖片的二值化處理
# 紅底變藍底
# 將在兩個閾值內(nèi)的像素值設(shè)置為白色（255），
#而不在閾值區(qū)間內(nèi)的像素值設(shè)置為黑色（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)，
#若是想將藍底變?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表示白色,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)

安啦！換證件照的小程序就寫好了。換上自己的圖片試試叭??！

以上就是使用python半分鐘輕松完成證件照換底色的詳細內(nèi)容，更多關(guān)于python完成證件照換底色的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: