python+openCV图像处理（十四）直方图反向投影

import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt

def back_projection_demo():#直方图反向投影
    image1 = cv.imread("1.png")
    image2 = cv.imread("2.png")
    roi_hsv = cv.cvtColor(image1, cv.COLOR_BGR2HSV)
    target_hsv = cv.cvtColor(image2, cv.COLOR_BGR2HSV)

    # show images
    cv.imshow("sample", image1)
    cv.imshow("target", image2)

    roiHist = cv.calcHist([roi_hsv], [0, 1], None, [32, 32], [0, 180, 0, 256])
    cv.normalize(roiHist, roiHist, 0, 255, cv.NORM_MINMAX)
    dst = cv.calcBackProject([target_hsv], [0, 1], roiHist, [0, 180, 0, 256], 1)
    cv.imshow("backProjectionDemo", dst)

def hist2d_demo(image):#2维直方图
    hsv = cv.cvtColor(image, cv.COLOR_BGR2HSV)
    #hist = cv.calcHist([image], [0, 1], None, [32, 32], [0, 180, 0, 256])
    hist = cv.calcHist([image], [0, 1], None, [180, 256], [0, 180, 0, 256])

    #cv.imshow("hist2d", hist)
    plt.imshow(hist, interpolation='nearest')
    plt.title("2D Histogram")
    plt.show()

src = cv.imread("1.png")
cv.namedWindow("input image", cv.WINDOW_AUTOSIZE)
cv.imshow("input image", src)

hist2d_demo(src)

back_projection_demo()
cv.waitKey(0)

cv.destroyAllWindows()