python實(shí)現(xiàn)證件照換底功能
更新時(shí)間：2019年08月20日 08:39:12 作者：yanni0616
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)證件照換底功能，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
本來(lái)是在找交通識(shí)別的程序，然后湊巧看見(jiàn)了證件照換底，于是學(xué)習(xí)了一下~一開(kāi)始在網(wǎng)上找了一個(gè)很普遍寫的程序，但是效果并不好，想要放棄了，然后看見(jiàn)了這個(gè)，參考：python opencv實(shí)現(xiàn)證件照換底功能
本人基于此做了一些改進(jìn)，便于自己運(yùn)行和調(diào)試，還加了中值濾波，本人覺(jué)得效果還不錯(cuò)~
代碼：
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() #加了這個(gè)后在圖片窗口按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])
以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
python MySQLdb Windows下安裝教程及問(wèn)題解決方法
這篇文章主要介紹了python MySQLdb Windows下安裝教程及問(wèn)題解決方法,本文講解了安裝數(shù)據(jù)庫(kù)mysql、安裝MySQLdb等步驟,需要的朋友可以參考下2015-05-05
基于OpenCV4.2實(shí)現(xiàn)單目標(biāo)跟蹤
這篇文章主要介紹了如何和何時(shí)使用OpenCV 4.2中可用的8種不同的跟蹤器- BOOSTING, MIL, KCF, TLD, MEDIANFLOW, GOTURN, MOSSE和CSRT，并用他們實(shí)現(xiàn)單目標(biāo)跟蹤，需要的可以參考一下2022-03-03
這篇文章主要介紹了python Plotly繪圖工具的簡(jiǎn)單使用，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-03-03
Python實(shí)現(xiàn)正則表達(dá)式匹配任意的郵箱方法
今天小編就為大家分享一篇Python實(shí)現(xiàn)正則表達(dá)式匹配任意的郵箱方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12
解決tensorflow打印tensor有省略號(hào)的問(wèn)題
今天小編就為大家分享一篇解決tensorflow打印tensor有省略號(hào)的問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-02-02
這篇文章主要介紹了Python用字典構(gòu)建多級(jí)菜單功能，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-07-07
深入理解Python的import機(jī)制有助于更好地組織代碼、提高代碼復(fù)用性,本文將深入研究Python的Import機(jī)制,包括模塊的導(dǎo)入過(guò)程、命名空間與作用域、相對(duì)導(dǎo)入以及包的結(jié)構(gòu)和導(dǎo)入等方面,通過(guò)豐富的示例代碼,助你更全面地理解和應(yīng)用這2024-01-01
今天小編就為大家分享一篇python 搜索大文件的實(shí)例代碼，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-07-07
Django rest framework工具包簡(jiǎn)單用法示例
這篇文章主要介紹了Django rest framework工具包簡(jiǎn)單用法,結(jié)合匿名訪問(wèn)控制的具體實(shí)例分析了Django rest framework工具包的注冊(cè)、路由設(shè)置、視圖、權(quán)限控制、配置等相關(guān)操作技巧,需要的朋友可以參考下2018-07-07
在paddle中安裝python-bidi出錯(cuò)問(wèn)題及解決
在Paddle中安裝python-bidi時(shí)遇到Rust和Cargo缺失的問(wèn)題,通過(guò)安裝舊版本的python-bidi（0.4.0）解決了問(wèn)題2025-02-02