1. 程式人生 > >opencv實現摳圖(單一背景),替換背景圖

opencv實現摳圖(單一背景),替換背景圖

下面簡單圖片演示一下:

提取mask:

===>

替換背景:

 + =

python的opencv程式碼如下:

# coding=utf-8

import cv2
import  numpy as np

img=cv2.imread('lp.jpg')
img_back=cv2.imread('back.jpg')
#日常縮放
rows,cols,channels = img_back.shape
img_back=cv2.resize(img_back,None,fx=0.7,fy=0.7)
cv2.imshow('img_back',img_back)

rows,cols,channels = img.shape
img=cv2.resize(img,None,fx=0.4,fy=0.4)
cv2.imshow('img',img)
rows,cols,channels = img.shape#rows,cols最後一定要是前景圖片的,後面遍歷圖片需要用到

#轉換hsv
hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
#獲取mask
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)

#遍歷替換
center=[50,50]#在新背景圖片中的位置
for i in range(rows):
    for j in range(cols):
        if dilate[i,j]==0:#0代表黑色的點
            img_back[center[0]+i,center[1]+j]=img[i,j]#此處替換顏色,為BGR通道
cv2.imshow('res',img_back)

cv2.waitKey(0)
cv2.destroyAllWindows()