Opencv抠图并面部检测后添加圣诞帽
从网上获取Trump的头像, 再使用EigenFaceRecognizer_create单独识别出Trump
import cv2
import numpy as np
filename='faces.jpg'
hat=cv2.imread('hat.jpg')
#构建mask
hsv=cv2.cvtColor(hat,cv2.COLOR_RGB2HSV)
black_lower=np.array([0,0,0])
black_upper=np.array([180,180,180])
mask=cv2.inRange(hsv,black_lower,black_upper)
#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)
def detect(filename):
face_cascade=cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
img=cv2.imread(filename)
gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
faces=face_cascade.detectMultiScale(gray,scaleFactor=1.055,minNeighbors=5,minSize=(30,30))
for (x,y,w,h) in faces:
#img=cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)#面部识别方框
sp=hat.shape
hatSizeH=int(sp[0]/sp[1]*w)
if hatSizeH>y:
hatSizeH=y
hatResize=cv2.resize(hat,(w,hatSizeH),interpolation=cv2.INTER_NEAREST)
dilateResize=cv2.resize(dilate,(w,hatSizeH),interpolation=cv2.INTER_NEAREST)
top=y-hatSizeH
if top<=0:
top=0
rows,cols,channels=hatResize.shape
roi=img[top:top+rows,x:x+cols]
for i in range(rows):
for j in range(cols):
if dilateResize[i,j]==0:#0代表黑色的点
img[top+i,x+j]=hatResize[i,j]#此处替换颜色,为BGR通道
cv2.namedWindow('leaders')
cv2.imshow('leaders',img)
cv2.imwrite('leaders_withHats_withoutRectangles.jpg',img)
cv2.waitKey(0)
detect(filename)
计组令人窒息, 丢下代码就跑, 有空再更