Python OpenCV Template Matching lỗi

Question

Tôi đã rối tung xung quanh với các ràng buộc OpenCV cho python trong một thời gian bây giờ và tôi muốn thử mẫu phù hợp, tôi nhận được lỗi này và tôi không có ý tưởng tại sao

C:\builds\master_PackSlaveAddon-win64-vc12-static\opencv\modules\imgproc\src\templmatch.cpp:910: error: (-215) (depth == CV_8U || depth == CV_32F) && type == _templ.type() && _img.dims() <= 2 in function cv::matchTemplate 

Bất cứ ai có bất kỳ manh mối nào về việc tại sao điều này có thể xảy ra không?

Source code:

import win32gui 
from PIL import ImageGrab 
import win32api, win32con 
import numpy 
deckVar = "deck.png" # Temporary 

def click(x,y): 
    win32api.SetCursorPos((x,y)) 
    win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN,x,y,0,0) 
    win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP,x,y,0,0) 

margin = 10 

def OOO(): # Order Of Operations 
    print None 

def main(): 
    deck = "test" 

    img = ImageGrab.grab() 

    HWNDHandle = win32gui.FindWindow(None, "Hearthstone"); 
    x,y,x2,y2 = win32gui.GetWindowRect(HWNDHandle) 
    print x,y,x2,y2 
    l = x2-x 
    h = y2-y 
    print l,h 

    img_recog(img,"imgs/my_collection.png") 

def img_recog(img,templ): 
    import cv2 
    import numpy as np 
    from matplotlib import pyplot as plt 


    img2 = numpy.array(img.getdata()).reshape(img.size[0], img.size[1], 3) 
    template = cv2.imread(templ,0) 
    w, h = template.shape[::-1] 

    # All the 6 methods for comparison in a list 
    methods = ['cv2.TM_CCOEFF', 'cv2.TM_CCOEFF_NORMED', 'cv2.TM_CCORR', 
       'cv2.TM_CCORR_NORMED', 'cv2.TM_SQDIFF', 'cv2.TM_SQDIFF_NORMED'] 


    img = img2.copy() 
    method = eval(methods[1]) 

    # Apply template Matching 
    try: 
     res = cv2.matchTemplate(img,template,method) 
    except Exception as e: 
     print str(e) 
     raw_input() 
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res) 


    # If the method is TM_SQDIFF or TM_SQDIFF_NORMED, take minimum 
    if method in [cv2.TM_SQDIFF, cv2.TM_SQDIFF_NORMED]: 
     top_left = min_loc 
    else: 
     top_left = max_loc 
    bottom_right = (top_left[0] + w, top_left[1] + h) 

    return cv2.rectangle(img,top_left, bottom_right, 255, 2) 


main() 

Answer 1

Chú ý đến thông báo lỗi:

error: (-215) (depth == CV_8U || depth == CV_32F) && type == _templ.type() && _img.dims() <= 2 in function cv::matchTemplate

Nó có nghĩa là kiểu dữ liệu của hình ảnh nên CV_8U hoặc CV_32F, và nó sẽ có 3 hoặc ít hơn các kênh truyền hình .

Nếu bạn không biết CV_8U, CV_32F có nghĩa là gì, hãy xem this list.

Có thể bạn đang truyền các đối tượng khó khăn khác với np.uint8 hoặc np.float32. bạn có thể dễ dàng chuyển đổi dtype NumPy đến 8-bit hoặc 32-bit sử dụng:

img.astype(np.float32) 
img.astype(np.uint8) 

Chỉ cần chú ý rằng OpenCV mong đợi CV_8U dữ liệu 8-bit để được trong khoảng 0..255 và CV_32F có thể ở bất kỳ phạm vi.