colordetection copy.py

import numpy as np
import cv2

cap = cv2.VideoCapture('lineDetect.mp4')


def findColor(frame, width, height):
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    upper_blue = np.array([208, 86, 49])
    lower_blue = np.array([208, 79, 30])

    mask = cv2.inRange(hsv, lower_blue, upper_blue)

    result = cv2.bitwise_and(frame, frame, mask=mask)

    return result


while True:
    ret, frame = cap.read()
    width = int(cap.get(3))
    height = int(cap.get(4))

    frame = findColor(frame, width, height)

    cv2.imshow('frame', frame)
    if cv2.waitKey(1) == ord('q'):
        break

cap.release()
cap.destroyAllWindows()