Better Hue overflow handling

2018-06-29 11:25:19 +02:00
parent fdc4ee6784
commit 9f3fe74be6
3 changed files with 13 additions and 31 deletions
--- a/pykick/utils.py
+++ b/pykick/utils.py
@@ -34,8 +34,8 @@ def imresize(frame, width=None, height=None):
 def hsv_mask(hsv, hsv_lower, hsv_upper):
    if hsv_lower[0] > hsv_upper[0]:
        mask_l = cv2.inRange(hsv, tuple(hsv_lower),
-                             tuple([180] + hsv_upper[1:]))
-        mask_u = cv2.inRange(hsv, tuple([0] + hsv_lower[1:]),
+                             (180,) + tuple(hsv_upper[1:]))
+        mask_u = cv2.inRange(hsv, (0,) + tuple(hsv_lower[1:]),
                             tuple(hsv_upper))
        return cv2.add(mask_l, mask_u)
    else: