ColorDetect_Red.py

# import the necessary packages
from picamera.array import PiRGBArray
from picamera import PiCamera
import time
import cv2
import numpy as np
# initialize the camera and grab a reference to the raw camera capture
camera = PiCamera()
camera.resolution = (640, 480)
camera.framerate = 32
rawCapture = PiRGBArray(camera, size=(640, 480))
# allow the camera to warmup
time.sleep(0.1)

# define the range of red
lower_red_0 = np.array([0, 70, 0]) 
upper_red_0 = np.array([5, 255, 255])
lower_red_1 = np.array([175, 70, 0]) 
upper_red_1 = np.array([180, 255, 255])

# capture frames from the camera
for frame in camera.capture_continuous(rawCapture, format="bgr", use_video_port=True):
	# grab the raw NumPy array representing the image, then initialize the timestamp
	# and occupied/unoccupied text
        image = frame.array
        hsv_img = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
        red_mask0 = cv2.inRange(hsv_img, lower_red_0, upper_red_0)
        red_mask1 = cv2.inRange(hsv_img, lower_red_1, upper_red_1)
        red_mask = cv2.bitwise_or(red_mask0, red_mask1)
	# show the frame
        cv2.imshow("Frame", image)
        cv2.imshow("Red Detection", red_mask)
        key = cv2.waitKey(1) & 0xFF
	# clear the stream in preparation for the next frame
        rawCapture.truncate(0)
	# if the `q` key was pressed, break from the loop
        if key == ord("q"):
                break

camera.close()
time.sleep(1.5)
cv2.destroyWindow("Frame")
cv2.destroyWindow("Red Detection")