Python OpenCV imshow预期参数



我正在尝试编写一个程序,显示给定窗口上的特定位置。当我运行代码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

最新更新