RPi + Python + OpenCV その３「クロマキー」

クロマキー合成

RPi 2BにUSBカメラを２台つないでクロマキー合成をやってみた。

#!/usr/bin/python
import cv2
import numpy as np
#capture frame size
width  = 320
height = 240
#chroma key color band HSV data
lower_color = np.array([204/2,100,100]) 
upper_color = np.array([230/2,255,255])
#OpenCV HSV H:0-180,S:0-255,V:0-255
#But populer HSV H:0-360 so I use [0..360]/2
#capture device open
cap = cv2.VideoCapture(0)   #for object
ret = cap.set(3,width)
ret = cap.set(4,height)
#
bak = cv2.VideoCapture(1)   #for background
ret = bak.set(3,width)
ret = bak.set(4,height)
while(1):
	# Take each frame
	ret, frame = cap.read()
	ret, back = bak.read()
	# Convert BGR to HSV
	hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
	# Threshold the HSV image to get only blue colors
	mask = cv2.inRange(hsv, lower_color, upper_color)
	inv_mask = cv2.bitwise_not(mask)
	# Bitwise-AND mask,inv_mask and original image
	res1 = cv2.bitwise_and(frame,frame,mask= inv_mask)
	res2 = cv2.bitwise_and(back,back,mask=  mask)
	#compsiting
	disp = cv2.bitwise_or(res1,res2,mask)
	#show
	cv2.imshow('frame',frame)
	cv2.imshow('disp',disp)
	cv2.imshow('back',back)
	# When hit the ESC-key go to exit
	k = cv2.waitKey(60) 
	if k == 27:
		break
#close all windows	
cv2.destroyAllWindows()

クロマキー合成

【参考 】RGBとHSVの相互変換確認　http://www.peko-step.com/tool/hsvrgb.html