OpenCV python 轮廓(连通域)匹配
图片[cs1.jpg]
图片[cs2.jpg]
图片[cc.jpg]
import cv2
def get_contours(img):
"""获取连通域
:param img: 输入图片
:return: 最大连通域
"""
# 灰度化, 二值化, 连通域分析
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, img_bin = cv2.threshold(img_gray, 127, 255, cv2.THRESH_BINARY)
img_contour, contours, hierarchy = cv2.findContours(img_bin, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
return contours[0]
def main():
# 1.导入图片
img_cs1 = cv2.imread("cs1.jpg")
img_cs2 = cv2.imread("cs2.jpg")
img_cc = cv2.imread("cc.jpg")
# 2.获取连通域
contour_cs1 = get_contours(img_cs1)
contour_cs2 = get_contours(img_cs2)
contour_cc = get_contours(img_cc)
# 3.计算匹配度
ret0 = cv2.matchShapes(contour_cs1, contour_cs1, 1, 0)
ret1 = cv2.matchShapes(contour_cs1, contour_cs2, 1, 0)
ret2 = cv2.matchShapes(contour_cs1, contour_cc, 1, 0)
# 4.打印匹配度与显示结果
print("相同的图像的 matchshapes:", ret0)
print("相似的图像的 matchshapes:", ret1)
print("不同的图像的 matchshapes:", ret2)
cv2.imshow("img_cs1", img_cs1)
cv2.imshow("img_cs2", img_cs2)
cv2.imshow("img_cc", img_cc)
cv2.waitKey()
cv2.destroyAllWindows()
if __name__ == '__main__':
main()
处理结果: