我正在尝试编写一个程序,显示给定窗口上的特定位置。当我运行代码OpenCV给我一个错误消息告诉我意外类型的参数垫。我不知道什么是错的,因为当我使用以下代码中的位置值它工作得很好。请友好,因为我对OpenCV相当陌生。
代码:
#Imports
import numpy as np
import cv2
import pytesseract
from PIL import ImageGrab
import win32gui
#Assign Tesseract and other Variables
winList = []
toplist = []
pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
lH = 33
lW = 20
#Window Handle Object
def enum_win(hwnd, result):
winText = win32gui.GetWindowText(hwnd)
winList.append((hwnd, winText))
win32gui.EnumWindows(enum_win, toplist)
typeHwnd = 0
#Find Window Handle
for (hwnd, winText) in winList:
if winText == "Program WinText Name Here":
typeHwnd = hwnd
if typeHwnd == 0:
print("Oops no Window Found")
while True:
position = win32gui.GetWindowRect(typeHwnd)
mid = (position[0] + (position[0] + position[2]))/2
sPos = [int(mid)-267, position[1] + 295, lW, lH]
screenshot = ImageGrab.grab(bbox=sPos)
screenshot = np.array(screenshot)
cv2.imshow("Screen", screenshot)
key = cv2.waitKey(25)
print(position)
print(mid)
错误信息:
Traceback (most recent call last):
File "C:/Users/MyName/PycharmProjects/Typing Bot/main.py", line 47, in <module>
cv2.imshow("Screen", screenshot)
TypeError: Expected Ptr<cv::UMat> for argument 'mat'
系统规格:
我正在运行Windows 10 64位和Python 3,并安装了所有导入模块的最新版本。
错误显示imshow
中的array
有问题,但您的问题始于grab
中的sPos
。
grab
需要bbox
,这意味着[x1, y1, x2, y2]
,但你使用[x, y, width, height]
,这开始问题,因为grab
返回图像大小(0,0)
和np.array()
不能转换它,所以它返回pillow image
而不是numpy array
,然后imshow
得到pillow image
而不是numpy array
,它不能显示它。
你必须使用[x, y, x+width, y+height]
最小工作示例-错误和正确的pos
import cv2
import numpy as np
import PIL.ImageGrab
x = 100
y = 100
w = 50
h = 50
pos = [x, y, w, h] # wrong
#pos = [x, y, x+w, y+h] # correct
print('bbox:', pos)
print('---')
img = PIL.ImageGrab.grab(pos)
print('img:', img)
print('type(img):', type(img))
print('img.size:', img.size)
print('---')
arr = np.array(img)
print('arr:', arr)
print('type(arr):', type(arr))
print('arr.size:', arr.size)
print('arr.shape:', arr.shape)
print('---')
cv2.imshow('test', arr)
print('Press ESC to exit')
while cv2.waitKey(100) != 27:
pass