pavel/kick-it
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Robuster colorpicker and detection demo

Browse Source
This commit is contained in:
Pavel Lutskov
2018-06-24 17:13:44 +02:00
parent 548e0d6fb5
commit 9c972c0a1c
3 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pykick/finders.py
View File

@@ -133,13 +133,18 @@ class GoalFinder(object):
class BallFinder(object):
def __init__(self, hsv_lower, hsv_upper, min_radius):
def __init__(self, hsv_lower, hsv_upper, min_radius=0.02):
self.hsv_lower = tuple(hsv_lower)
self.hsv_upper = tuple(hsv_upper)
self.min_radius = min_radius
self.history = deque(maxlen=64)
def primary_mask(self, frame):
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
mask = cv2.inRange(hsv, self.hsv_lower, self.hsv_upper)
return mask
def find(self, frame):
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)