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
Другие вопросы по тегам