Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- """
- Created on Fri Dec 19 22:43:24 2014
- """
- import cv2
- import numpy as np
- def nothing(x): #needed for the trackbars
- pass
- cv2.namedWindow('Calibration')
- # create trackbars for color change
- cv2.createTrackbar('Hmin','Calibration',0,179,nothing)
- cv2.createTrackbar('Hmax','Calibration',0,179,nothing)
- cv2.createTrackbar('Smin','Calibration',0,255,nothing)
- cv2.createTrackbar('Smax','Calibration',0,255,nothing)
- cv2.createTrackbar('Vmin','Calibration',0,255,nothing)
- cv2.createTrackbar('Vmax','Calibration',0,255,nothing)
- webcam = cv2.VideoCapture(0)
- webcam.grab()
- while(1):
- k = cv2.waitKey(1) & 0xFF
- if k == 27:
- break
- ret, frame = webcam.retrieve(channel=0)
- calibration_hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
- # get current positions of trackbars
- hmin = min(cv2.getTrackbarPos('Hmin','Calibration'),cv2.getTrackbarPos('Hmax','Calibration'))
- hmax = max(cv2.getTrackbarPos('Hmax','Calibration'),cv2.getTrackbarPos('Hmin','Calibration'))
- smin = min(cv2.getTrackbarPos('Smin','Calibration'),cv2.getTrackbarPos('Smax','Calibration'))
- smax = max(cv2.getTrackbarPos('Smax','Calibration'),cv2.getTrackbarPos('Smin','Calibration'))
- vmin = min(cv2.getTrackbarPos('Vmin','Calibration'),cv2.getTrackbarPos('Vmax','Calibration'))
- vmax = max(cv2.getTrackbarPos('Vmax','Calibration'),cv2.getTrackbarPos('Vmin','Calibration'))
- # Threshold the HSV image to get only selected color
- lower_thr = np.array([hmin, smin, vmin], dtype=np.uint8)
- upper_thr = np.array([hmax,smax,vmax], dtype=np.uint8)
- mask = cv2.inRange(calibration_hsv, lower_thr, upper_thr)
- cv2.imshow('Calibration',mask)
- webcam.release()
- cv2.destroyAllWindows()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement