python實(shí)現(xiàn)證件照換底功能
更新時間：2019年08月20日 08:39:12 作者：yanni0616
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)證件照換底功能，具有一定的參考價值，感興趣的小伙伴們可以參考一下
本來是在找交通識別的程序，然后湊巧看見了證件照換底，于是學(xué)習(xí)了一下~一開始在網(wǎng)上找了一個很普遍寫的程序，但是效果并不好，想要放棄了，然后看見了這個，參考：python opencv實(shí)現(xiàn)證件照換底功能
本人基于此做了一些改進(jìn)，便于自己運(yùn)行和調(diào)試，還加了中值濾波，本人覺得效果還不錯~
代碼：
import cv2
import numpy as np
def cvtBackground(path,color):
"""
功能：給證件照更換背景色（常用背景色紅、白、藍(lán)）
輸入?yún)?shù)：path:照片路徑
color:背景色 <格式[B,G,R]>
"""
im=cv2.imread(path)
im_hsv=cv2.cvtColor(im,cv2.COLOR_BGR2HSV) #BGR和HSV的轉(zhuǎn)換使用 cv2.COLOR_BGR2HSV
#aim=np.uint8([[im[0,0,:]]])
#hsv_aim=cv2.cvtColor(aim,cv2.COLOR_BGR2HSV)
mask=cv2.inRange(im_hsv,np.array([im_hsv[0,0,0]-5,100,100]),np.array([im_hsv[0,0,0]+5,255,255])) #利用cv2.inRange函數(shù)設(shè)閾值，去除背景部分
mask1=mask #在lower_red～upper_red之間的值變成255
img_median = cv2.medianBlur(mask, 5) #自己加，中值濾波，去除一些邊緣噪點(diǎn)
mask = img_median
mask_inv=cv2.bitwise_not(mask)
img1=cv2.bitwise_and(im,im,mask=mask_inv) #將人物摳出
bg=im.copy()
rows,cols,channels=im.shape
bg[:rows,:cols,:]=color
img2=cv2.bitwise_and(bg,bg,mask=mask) #將背景底板摳出
img=cv2.add(img1,img2)
image={'im':im,'im_hsv':im_hsv,'mask':mask1,'img':img,'img_median':img_median}
cv2.startWindowThread() #加了這個后在圖片窗口按Esc就可以關(guān)閉圖片窗口
for key in image:
cv2.namedWindow(key)
cv2.imshow(key,image[key])
cv2.waitKey(0)
cv2.destroyAllWindows()
return img
#test
if __name__=='__main__':
img=cvtBackground('3.jpg',[0,0,180])
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
Python采用Django開發(fā)自己的博客系統(tǒng)
這篇文章主要為大家詳細(xì)介紹了Python采用Django開發(fā)自己的博客系統(tǒng)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2016-08-08
Python+OpenCV+圖片旋轉(zhuǎn)并用原底色填充新四角的例子
今天小編就為大家分享一篇Python+OpenCV+圖片旋轉(zhuǎn)并用原底色填充新四角的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
在本篇文章里小編給大家整理的是一篇關(guān)于python中zip()函數(shù)遍歷多個列表方法，對此有興趣的朋友們可以學(xué)習(xí)下。2021-02-02
python讀取pdf格式文檔的實(shí)現(xiàn)代碼
這篇文章主要給大家介紹了關(guān)于python讀取pdf格式文檔的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04
python線程池 ThreadPoolExecutor 的用法示例
這篇文章主要介紹了python線程池 ThreadPoolExecutor 的用法示例，幫助大家更好得理解和使用python線程池，感興趣的朋友可以了解下2020-10-10
使用python將csv數(shù)據(jù)導(dǎo)入mysql數(shù)據(jù)庫
這篇文章主要為大家詳細(xì)介紹了如何使用python將csv數(shù)據(jù)導(dǎo)入mysql數(shù)據(jù)庫,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-05-05
因?yàn)閷?shí)驗(yàn)需要,實(shí)現(xiàn)了一下寫入txt文件,下面這篇文章主要給大家介紹了關(guān)于如何利用Python將list列表寫入文件并讀取的幾種方法,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-03-03
python利用logging模塊實(shí)現(xiàn)根據(jù)日志級別打印不同顏色日志的代碼案例
這篇文章主要介紹了python利用logging模塊實(shí)現(xiàn)根據(jù)日志級別打印不同顏色日志,本文通過實(shí)例代碼給大家詳細(xì)講解，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-12-12
Scrapy，Python開發(fā)的一個快速,高層次的屏幕抓取和web抓取框架，用于抓取web站點(diǎn)并從頁面中提取結(jié)構(gòu)化的數(shù)據(jù)。Scrapy用途廣泛，可以用于數(shù)據(jù)挖掘、監(jiān)測和自動化測試。2017-10-10
本文通過實(shí)例代碼給大家介紹了python ftp兩個文件夾間的同步，代碼簡單易懂，非常不錯，具有一定的參考借鑒價值，需要的朋友參考下吧2018-05-05