使用功能matchTemplate的OpenCV错误

时间:2019-04-21 16:26:21

标签: numpy opencv image-processing

在OpenCV中使用matchTemplate函数时,出现以下错误:模板图像大于原始图像。如何克服呢?

代码如下:

def imagecheck(name1):
  os.chdir('/content/drive/My Drive/Mad Street Den/Images')
  main_image = cv2.imread('image_name_100.jpg')
  gray_image = cv2.cvtColor(main_image, cv2.COLOR_BGR2GRAY)

  #open the template as gray scale image
  os.chdir('/content/drive/My Drive/Mad Street Den/Crops')
  template = cv2.imread(name1, 0)
  width, height = template.shape[::-1] #get the width and height

  #match the template using cv2.matchTemplate
  match = cv2.matchTemplate(gray_image, template, cv2.TM_CCOEFF_NORMED)
  threshold = 0.9
  position = np.where(match >= threshold) #get the location of template in the image

  for point in zip(*position[::-1]): #draw the rectangle around the matched template
     cv2.rectangle(main_image, point, (point[0] + width, point[1] + height), (0, 204, 153), 2)
  #result=[position[1][0],position[0][0],position[0][1],position[0][2]]

  result=[]
  if (all (position)):
    result.append(int(position[1]))
    result.append(int(position[0]))
    result.append(int(position[1]+width))
    result.append(int(position[0]+height))
  return (result)
  #cv2_imshow(main_image)
  
for i in range(0,273):
  name1='image_name_'+str(i)+'.jpg'
  result=imagecheck(name1)
  print(name1, ' : ',result)
  

错误是

error: OpenCV(3.4.3) /io/opencv/modules/imgproc/src/templmatch.cpp:1107: error: (-215:Assertion failed) _img.size().height <= _templ.size().height && _img.size().width <= _templ.size().width in function 'matchTemplate' site:stackoverflow.com

1 个答案:

答案 0 :(得分:0)

如果模板较大,则不尝试将模板与图像进行匹配,可以避免此问题。将模板尺寸与图像尺寸进行比较,在这种情况下,将return []与模板尺寸进行比较。