基于OpenCV python3實現(xiàn)證件照換背景的方法

更新時間：2019年03月22日 09:52:48 作者：ahaofan

這篇文章主要介紹了基于OpenCV python3實現(xiàn)證件照換背景的方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

簡述

生活中經(jīng)常要用到各種要求的證件照電子版，紅底，藍(lán)底，白底等，大部分情況我們只有其中一種，所以通過技術(shù)手段進(jìn)行合成，用ps處理證件照，由于技術(shù)不到位，有瑕疵，所以想用python&openCV通過代碼的方式實現(xiàn)背景顏色替換，加強(qiáng)一下對于openCV的學(xué)習(xí)，鍛煉一下編碼水平。

軟件環(huán)境：

python3.5
opencv2
windows 10

圖像載入

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

import cv2
import numpy as np

img=cv2.imread('zjz.jpg')

由于證件照太大，不方便顯示，故進(jìn)行縮放

#縮放
rows,cols,channels = img.shape
img=cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
cv2.imshow('img',img)

原圖如下

（圖片源于網(wǎng)絡(luò)，已經(jīng)馬賽克處理，如有侵權(quán)，私信立即刪除）

獲取背景區(qū)域

首先將讀取的圖像默認(rèn)BGR格式轉(zhuǎn)換為HSV格式，然后通過inRange函數(shù)獲取背景的mask。
HSV顏色范圍參數(shù)可調(diào)節(jié)根據(jù)這篇文章

hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
lower_blue=np.array([78,43,46])
upper_blue=np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)
cv2.imshow('Mask', mask)

獲得的mask如下圖

如圖所示藍(lán)色的背景在圖中用白色表示，白色區(qū)域就是要替換的部分，但是黑色區(qū)域內(nèi)有白點干擾，所以進(jìn)一步優(yōu)化。
腐蝕和膨脹

#腐蝕膨脹
erode=cv2.erode(mask,None,iterations=1)
cv2.imshow('erode',erode)
dilate=cv2.dilate(erode,None,iterations=1)
cv2.imshow('dilate',dilate)

經(jīng)過腐蝕和膨脹操作后如下圖

處理后圖像單獨白色點消失。

替換背景色

遍歷全部像素點，如果該顏色為dilate里面為白色（255）則說明該點所在背景區(qū)域，于是在原圖img中進(jìn)行顏色替換。

#遍歷替換
for i in range(rows):
 for j in range(cols):
  if dilate[i,j]==255:
   img[i,j]=(0,0,255)#此處替換顏色，為BGR通道
cv2.imshow('res',img)

最終結(jié)果如下

（圖片源于網(wǎng)絡(luò)，已經(jīng)馬賽克處理，如有侵權(quán)，私信立即刪除）

總結(jié)

最開始想直接通過遍歷全圖進(jìn)行替換背景色，但是圖像中難免有些像素點和背景色一樣，造成了干擾，導(dǎo)致最后結(jié)果不盡人意，所以想通過這種方法進(jìn)行處理。顯然最后有明顯的ps痕跡。

最后貼上完整代碼，不足之處歡迎各位指正！

import cv2
import numpy as np

img=cv2.imread('zjz.jpg')
#縮放
rows,cols,channels = img.shape
img=cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
cv2.imshow('img',img)

#轉(zhuǎn)換hsv
hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
lower_blue=np.array([78,43,46])
upper_blue=np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)
cv2.imshow('Mask', mask)

#腐蝕膨脹
erode=cv2.erode(mask,None,iterations=1)
cv2.imshow('erode',erode)
dilate=cv2.dilate(erode,None,iterations=1)
cv2.imshow('dilate',dilate)

#遍歷替換
for i in range(rows):
 for j in range(cols):
  if dilate[i,j]==255:
   img[i,j]=(0,0,255)#此處替換顏色，為BGR通道
cv2.imshow('res',img)

cv2.waitKey(0)
cv2.destroyAllWindows()

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: