OpenCV python 轮廓(连通域)匹配

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()

处理结果: