使用模板匹配方法检测苹果缺陷

“”"
使用模板匹配方法检测苹果缺陷
“”"

import cv2
import matplotlib.pyplot as plt

 #读取待检测苹果图像
img = cv2.imread("image/apple_normal.bmp",0)
#读取苹果缺陷模板图像
template = cv2.imread("image/apple_defect.bmp",0)
#获得苹果缺陷模板图像的高度和宽度
height,width = template.shape[::]

#以方差为依据进行匹配
rv = cv2.matchTemplate(img,template,cv2.TM_SQDIFF)
#匹配函数的返回值
minVal,maxVal,minLoc,maxLoc = cv2.minMaxLoc(rv)
#查找最小值所在位置(值越小匹配度越好)，topLeft为模板匹配位置的左上角坐标
topLeft = minLoc
#bottomRight为模板匹配位置的右下角坐标
bottomRight = (topLeft[0] + width,topLeft[1] + height)
#将苹果缺陷的模板匹配位置用矩形标注出来，255表示矩形为白色，1表示矩形宽度
cv2.rectangle(img,topLeft,bottomRight,255,1)

#模板匹配结果
plt.subplot(),plt.imshow(img,cmap='gray')
plt.title('Detected Point'),plt.xticks([]),plt.yticks([])
plt.show()

[版权声明：本文为博主原创文章，如要转载请在转载的文章后附上本人博客链接]https://editor.csdn.net/md?articleId=103892177