import cv2

import numpy as np

def nothing(x):

pass

cap = cv2.VideoCapture(0);

# 사용자가 색상을 조절하여 쉽게 추출할 수 있는 트래킹 창 만들기

cv2.namedWindow("Tracking")

cv2.createTrackbar("LH", "Tracking", 0, 255, nothing)

cv2.createTrackbar("LS", "Tracking", 0, 255, nothing)

cv2.createTrackbar("LV", "Tracking", 0, 255, nothing)

cv2.createTrackbar("UH", "Tracking", 255, 255, nothing)

cv2.createTrackbar("US", "Tracking", 255, 255, nothing)

cv2.createTrackbar("UV", "Tracking", 255, 255, nothing)

while True:

#frame = cv2.imread('./img/smarties.png')

_, frame = cap.read()

hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) # BGR 이미지를 HSV로 변경

l_h = cv2.getTrackbarPos("LH", "Tracking")

l_s = cv2.getTrackbarPos("LS", "Tracking")

l_v = cv2.getTrackbarPos("LV", "Tracking")

u_h = cv2.getTrackbarPos("UH", "Tracking")

u_s = cv2.getTrackbarPos("US", "Tracking")

u_v = cv2.getTrackbarPos("UV", "Tracking")

l_b = np.array([l_h,l_s,l_v])

u_b = np.array([u_h,u_s,u_v])

mask = cv2.inRange(hsv, l_b, u_b) # hsv에서 해당 색상 영역만 캐치

res = cv2.bitwise_and(frame, frame, mask=mask) # frame에서 마스크로 영역 설정한 부분만 캐치

cv2.imshow("frame", frame)

cv2.imshow("mask", mask)

cv2.imshow("res", res)

key = cv2.waitKey(1)

if key == 27:

break

cap.release()

cv2.destroyAllWindows()