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'
Технические характеристики системы:
Я использую 64-разрядную версию Windows 10 и Python 3 с самой последней версией всех импортированных модулей.
1 ответ
Ошибка показывает проблему с
array
но ваша проблема начинается с
sPos
в .
потребности
bbox
что значит
[x1, y1, x2, y2]
но вы используете
[x, y, width, height]
и это запускает проблему, потому что
grab
возвращает изображение с размером
(0,0)
и
np.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