Saner color calibration. Finder API changed

This commit is contained in:
2018-06-23 13:08:12 +02:00
parent 2fff7d19a2
commit 126329bafd
6 changed files with 129 additions and 104 deletions

View File

@@ -29,23 +29,3 @@ def imresize(frame, width=None, height=None):
sf = 0
sz = (width, height)
return cv2.resize(frame, sz, fx=sf, fy=sf)
def field_mask(frame, hsv_lower, hsv_upper):
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
blurred = cv2.GaussianBlur(hsv, (25, 25), 20)
thr = cv2.inRange(blurred, tuple(hsv_lower), tuple(hsv_upper))
thr = cv2.erode(thr, None, iterations=4)
thr = cv2.dilate(thr, None, iterations=8)
cnts, _ = cv2.findContours(thr.copy(), cv2.RETR_EXTERNAL,
cv2.CHAIN_APPROX_SIMPLE)
field = max(cnts, key=cv2.contourArea)
field = cv2.convexHull(field)
# print(field)
mask = np.zeros(thr.shape, dtype=np.uint8)
print(mask.dtype)
thr = cv2.cvtColor(thr, cv2.COLOR_GRAY2BGR)
cv2.drawContours(mask, (field,), -1, 255, -1)
# The ususal
return mask