我已经在Python的OpenCV绑定上折腾了一段时间,我想尝试模板匹配,我得到了这个错误,我不知道为什么
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
有人知道为什么会发生这种事吗?
源代码:
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()
2条答案
按热度按时间7xllpg7q1#
请注意错误消息:
误差:(-215)(深度== CV_8U|| depth == CV_32F)&& type ==_templ. type()&&_img. dims()<= 2在函数cv::matchTemplate中
这意味着图像的数据类型应为CV_8U或CV_32F,且其应具有3个或更少的通道。
如果你不知道CV_8U,CV_32F是什么意思,请参阅此列表。
可能你传递的是numpy对象,而不是np.uint8或np.float32。您可以轻松地将numpy dtype转换为8位或32位使用:
请注意,OpenCV期望CV_8U 8位数据在0..255范围内,CV_32F可以在任何范围内。
oaxa6hgo2#
我只是把png图片改成png类型,它就通过了。