-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcolor.py
36 lines (27 loc) · 1013 Bytes
/
color.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import cv2
import numpy as np
cam = cv2.VideoCapture(0)
cv2.namedWindow('Colour Detection')
def window(x):
pass
cv2.createTrackbar('Hue', 'Colour Detection', 0, 179, window)
cv2.createTrackbar('Saturation', 'Colour Detection', 0, 255, window)
cv2.createTrackbar('Value', 'Colour Detection', 0, 255, window)
while(True):
ret, img = cam.read()
img = np.flip(img,axis=1)
img = cv2.resize(img, (480,360))
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
blurred = cv2.GaussianBlur(hsv, (11, 11), 0)
h = cv2.getTrackbarPos('Hue', 'Colour Detection')
s = cv2.getTrackbarPos('Saturation', 'Colour Detection')
v = cv2.getTrackbarPos('Value', 'Colour Detection')
lower_colour = np.array([h, s, v])
upper_colour = np.array([180, 255, 255])
mask = cv2.inRange(hsv, lower_colour, upper_colour)
cv2.imshow('Colour Detection', cv2.bitwise_and(img, img, mask=mask))
key = cv2.waitKey(1) & 0xFF
if key == ord("q"):
break
cam.release()
cv2.destroyAllWindows()