OpenCV图像相似度检测,Python实现

OpenCV图像相似度检测,简单的说是大图中找小图,以图找图。

假设把大图

 

中的一部分图Android小机器人检测出来

并用红色的线框出来。

# OpenCV图像相似度检测matchTemplate,Python实现
import cv2 as cv

if __name__ == "__main__":
    # 先把图片灰度处理。
    img = cv.imread('pic.jpg')
    img_gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)

    template = cv.imread('t.jpg')
    template_gray = cv.cvtColor(template, cv.COLOR_BGR2GRAY)
    h, w = template.shape[:2]

    # 匹配
    result = cv.matchTemplate(img_gray, template_gray, cv.TM_CCOEFF_NORMED)

    min_val, max_val, min_loc, max_loc = cv.minMaxLoc(result)
    print(max_val)
    print(min_val)

    # max_loc为左上角
    # 右下角
    right_bottom = (max_loc[0] + w, max_loc[1] + h)

    # 画矩形,红色的线框出来。
    cv.rectangle(img=img, pt1=max_loc, pt2=right_bottom, color=(0, 0, 255), thickness=3)

    cv.imshow('result', img)
    cv.waitKey(0)
    cv.destroyAllWindows()

运行结果:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章