Блог

 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)