-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathobjectDetect.py
51 lines (36 loc) · 1.15 KB
/
objectDetect.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
37
38
39
40
41
42
43
44
45
46
47
import cv2 as cv
import numpy as np
def nothing(c):
pass
cap=cv.VideoCapture(0)
cv.namedWindow('Tracking')
cv.createTrackbar("LH",'Tracking',0,255,nothing)
cv.createTrackbar("UH",'Tracking',255,255,nothing)
cv.createTrackbar("LS",'Tracking',0,255,nothing)
cv.createTrackbar("US",'Tracking',255,255,nothing)
cv.createTrackbar("LV",'Tracking',0,255,nothing)
cv.createTrackbar("UV",'Tracking',255,255,nothing)
while True:
#frame=cv.imread('smarties.png')
_, frame=cap.read()
hsv=cv.cvtColor(frame,cv.COLOR_BGR2HSV)
l_h=cv.getTrackbarPos("LH",'Tracking')
u_h=cv.getTrackbarPos("UH",'Tracking')
l_s=cv.getTrackbarPos("LS",'Tracking')
u_s=cv.getTrackbarPos("US",'Tracking')
l_v=cv.getTrackbarPos("LV",'Tracking')
u_v=cv.getTrackbarPos("UV",'Tracking')
#l_b=np.array([110,50,50])
#u_b=np.array([130,255,255])
l_b=np.array([l_h,l_s,l_v])
u_b=np.array([u_h,u_s,u_v])
mask=cv.inRange(hsv,l_b,u_b)
res=cv.bitwise_and(frame,frame,mask=mask)
cv.imshow("frame",frame)
cv.imshow('res',res)
cv.imshow('mask',mask)
key=cv.waitKey(1)
if key==27:
break
cap.release()
cv.destroyAllWindows()