python 模板匹配多个物体

参考：

https://www.jianshu.com/p/c20adfa72733

• 平方差匹配CV_TM_SQDIFF：用两者的平方差来匹配，最好的匹配值为0
• 归一化平方差匹配CV_TM_SQDIFF_NORMED
• 相关匹配CV_TM_CCORR：用两者的乘积匹配，数值越大表明匹配程度越好
• 归一化相关匹配CV_TM_CCORR_NORMED
• 相关系数匹配CV_TM_CCOEFF：用两者的相关系数匹配，1表示完美的匹配，-1表示最差的匹配
• 归一化相关系数匹配CV_TM_CCOEFF_NORMED

  

   

    

     

    
    

     

      匹配多个物体
     
    
   

    

     

    
    

     

      前面我们是找最大匹配的点，所以只能匹配一次。我们可以设定一个匹配阈值来匹配多次：
     
    
   

    

     

    
    

     

      # 1.读入原图和模板
     
    
   

    

     

    
    

     

      img_rgb = cv2.imread('mario.jpg')
     
    
   

    

     

    
    

     

      img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
     
    
   

    

     

    
    

     

      template = cv2.imread('mario_coin.jpg', 0)
     
    
   

    

     

    
    

     

      h, w = template.shape[:2]
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # 2.标准相关模板匹配
     
    
   

    

     

    
    

     

      res = cv2.matchTemplate(img_gray, template, cv2.TM_CCOEFF_NORMED)
     
    
   

    

     

    
    

     

      threshold = 0.8 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # 3.这边是Python/Numpy的知识，后面解释
     
    
   

    

     

    
    

     

      loc = np.where(res >= threshold)  # 匹配程度大于%80的坐标y,x
     
    
   

    

     

    
    

     

      for pt in zip(*loc[::-1]):  # *号表示可选参数
     
    
   

    

     

    
    

     

       right_bottom = (pt[0] + w, pt[1] + h)
     
    
   

    

     

    
    

     

       cv2.rectangle(img_rgb, pt, right_bottom, (0, 0, 255), 2)
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/95319712