py+cv+hsv抓特征

# -*- coding: UTF-8 -*-

import cv2 as cv
import numpy as np

import win32api, win32con, win32gui

from PIL import Image, ImageGrab

def get_window_pos(name):
    name = name
    handle = win32gui.FindWindow(0, name)
    if handle == 0:
        return 0, 0, 0, 0
    else:
        return win32gui.GetWindowRect(handle)
        
x1, y1, x2, y2 = get_window_pos("xxx")
 

print(x1,y1,x2,y2)

#img_pil = ImageGrab.grab((x1, y1, x2, y2))
#img_pil = np.array(img_pil)
#img_cv = cv.cvtColor(img_pil, cv.COLOR_RGB2BGR)
#cv.imshow("aa", img_cv)
#cv.waitKey(0);

#exit()

img_org = cv.imread("1.bmp", cv.IMREAD_COLOR)

img_hsv = cv.cvtColor(img_org, cv.COLOR_BGR2HSV)

cv.imshow("img_org", img_org)

cv.imshow("img_hsv", img_hsv)

cv.namedWindow("track bar", cv.WINDOW_FREERATIO);

def cb(n):
    low_ = np.array([cv.getTrackbarPos("H-L", "track bar"), cv.getTrackbarPos("S-L", "track bar"), cv.getTrackbarPos("V-L", "track bar")])
    high_ = np.array([cv.getTrackbarPos("H-H", "track bar"), cv.getTrackbarPos("S-H", "track bar"), cv.getTrackbarPos("V-H", "track bar")])
    img_hsv_filt = cv.inRange(img_hsv, low_, high_)
    img_filt = cv.bitwise_and(img_org, img_org, mask=img_hsv_filt)
    cv.imshow("img_filt", img_filt)

cv.createTrackbar("H-H", "track bar", 0, 180, cb)
cv.createTrackbar("S-H", "track bar", 0, 255, cb)
cv.createTrackbar("V-H", "track bar", 0, 255, cb)

cv.createTrackbar("H-L", "track bar", 0, 180, cb)
cv.createTrackbar("S-L", "track bar", 0, 255, cb)
cv.createTrackbar("V-L", "track bar", 0, 255, cb)

cv.imshow("track bar" ,img_org)

state = 1

while 1:
    key = cv.waitKey(100)
    if key == ord('s'):
        state = 0
    elif key == ord('c'):
        state = 1
    elif key == ord('q'):
        exit()
    img_pil = ImageGrab.grab((x1, y1, x2, y2))
    img_pil = np.array(img_pil)
    img_cv = cv.cvtColor(img_pil, cv.COLOR_RGB2BGR)
    if state == 1:
        img_org = img_cv
        img_hsv = cv.cvtColor(img_org, cv.COLOR_BGR2HSV)
        cb(2)