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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
| import cv2
import numpy as np
import time
mode = 'camera'
if mode == 'camera':
cap = cv2.VideoCapture(0)
def update(x):
global gs, erode, Hmin, Smin, Vmin, Hmax, Smax, Vmax, img, Hmin2, Hmax2, img0, size_min
if mode == 'camera':
ret, img0 = cap.read()
elif mode == 'picture':
img0 = cv2.imread('test.jpg')
img = img0.copy()
gs = cv2.getTrackbarPos('gs', 'image')
erode = cv2.getTrackbarPos('erode', 'image')
Hmin = cv2.getTrackbarPos('Hmin1', 'image')
Smin = cv2.getTrackbarPos('Smin', 'image')
Vmin = cv2.getTrackbarPos('Vmin', 'image')
Hmax = cv2.getTrackbarPos('Hmax1', 'image')
Smax = cv2.getTrackbarPos('Smax', 'image')
Vmax = cv2.getTrackbarPos('Vmax', 'image')
Hmin2 = cv2.getTrackbarPos('Hmin2', 'image')
Hmax2 = cv2.getTrackbarPos('Hmax2', 'image')
size_min = cv2.getTrackbarPos('size_min', 'image')
gs_frame = cv2.GaussianBlur(img, (gs, gs), 1)
hsv = cv2.cvtColor(gs_frame, cv2.COLOR_BGR2HSV)
erode_hsv = cv2.erode(hsv, None, iterations=erode)
inRange_hsv = cv2.inRange(erode_hsv, np.array([Hmin, Smin, Vmin]), np.array([Hmax, Smax, Vmax]))
inRange_hsv2 = cv2.inRange(erode_hsv, np.array([Hmin2, Smin, Vmin]), np.array([Hmax2, Smax, Vmax]))
img = inRange_hsv + inRange_hsv2
cnts = cv2.findContours(img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)[-2]
target_list = []
pos = []
if size_min < 1:
size_min = 1
for c in cnts:
if cv2.contourArea(c) < size_min:
continue
else:
target_list.append(c)
for cnt in target_list:
x, y, w, h = cv2.boundingRect(cnt)
cv2.rectangle(img0, (x, y), (x + w, y + h), (0, 255, 0), 2)
pos.append([int(x + w / 2), y + h / 2])
print(pos)
def img_test():
sleep = 0.1
gs = 0
erode = 0
Hmin1 = 100
Hmax1 = 125
Hmin2 = 179
Hmax2 = 0
Smin = 130
Smax = 255
Vmin = 50
Vmax = 240
size_min = 1000
cv2.namedWindow('image', cv2.WINDOW_NORMAL)
cv2.createTrackbar('gs', 'image', 0, 8, update)
cv2.createTrackbar('erode', 'image', 0, 8, update)
cv2.createTrackbar('Hmin1', 'image', 0, 179, update)
cv2.createTrackbar('Hmax1', 'image', 0, 179, update)
cv2.createTrackbar('Hmin2', 'image', 0, 179, update)
cv2.createTrackbar('Hmax2', 'image', 0, 179, update)
cv2.createTrackbar('Smin', 'image', 0, 255, update)
cv2.createTrackbar('Smax', 'image', 0, 255, update)
cv2.createTrackbar('Vmin', 'image', 0, 255, update)
cv2.createTrackbar('Vmax', 'image', 0, 255, update)
cv2.createTrackbar('size_min', 'image', 1, 100000, update)
cv2.setTrackbarPos('gs', 'image', gs)
cv2.setTrackbarPos('erode', 'image', erode)
cv2.setTrackbarPos('Hmin1', 'image', Hmin1)
cv2.setTrackbarPos('Hmax1', 'image', Hmax1)
cv2.setTrackbarPos('Hmin2', 'image', Hmin2)
cv2.setTrackbarPos('Hmax2', 'image', Hmax2)
cv2.setTrackbarPos('Smin', 'image', Smin)
cv2.setTrackbarPos('Smax', 'image', Smax)
cv2.setTrackbarPos('Vmin', 'image', Vmin)
cv2.setTrackbarPos('Vmax', 'image', Vmax)
cv2.setTrackbarPos('size_min', 'image', size_min)
while (True):
try:
update(1)
except:
pass
cv2.imshow('image', img)
cv2.imshow('image1', img)
cv2.imshow('image0', img0)
time.sleep(sleep)
if cv2.waitKey(1) == 27:
break
cv2.destroyAllWindows()
if __name__ == '__main__':
img_test()
|
