Python OpenCV模板匹配错误

问题描述

一段时间以来，我一直在处理python的OpenCV绑定，我想尝试进行模板匹配，但出现此错误，我也不知道为什么

I've been messing around with the OpenCV bindings for python for a while now and i wanted to try template matching, i get this error and i have no idea why

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

任何人都知道为什么会发生这种情况吗?

Anyone have any clues as to why this might be happening?

源代码:

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()

推荐答案

请注意错误消息:

这意味着图像的数据类型应为CV_8U或CV_32F，并且应具有3个或更少的通道.

It means the data type of the image should be CV_8U or CV_32F, and it should have 3 or less channels.

如果您不知道CV_8U，CV_32F是什么意思，请参见此列表.

If you don't know what CV_8U, CV_32F means, see this list.

可能您正在传递np.uint8或np.float32以外的numpy对象.您可以使用以下命令轻松地将numpy dtype转换为8位或32位:

Probably you are passing numpy objects other than np.uint8 or np.float32.you can easily convert numpy dtype to 8-bit or 32-bit using:

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

请注意，OpenCV期望CV_8U 8位数据的范围是0..255，而CV_32F的范围可以是任何范围.

Just pay attention that OpenCV expect CV_8U 8-bit data to be in the range 0..255 and CV_32F can be in any range.

这篇关于Python OpenCV模板匹配错误的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！