《python+opencv3.3视频教学 基础入门》直方图反向投影 笔记

源码：

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


def back_projection_demo():
    sample = cv.imread("C:/cvtest/0002.jpg")
    target = cv.imread("C:/cvtest/0000.jpg")
    roi_hsv = cv.cvtColor(sample, cv.COLOR_BGR2HSV)
    target_hsv = cv.cvtColor(target, cv.COLOR_BGR2HSV)

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

    # roiHist = cv.calcHist([roi_hsv], [0, 1], None, [180, 256], [0, 180, 0, 256])
    # 直方图bins越多，对每个像素细分的越厉害，细分越厉害就会导致反向投影
    # 的结果会产生碎片化，所以我们可以不用分的那么细，我们就分的稍微少一点。
    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, 255], 1)

    dst = cv.cvtColor(dst, cv.COLOR_GRAY2BGR)
    dst = cv.bitwise_and(dst, target)

    cv.imshow("backProjectionDemo", dst)


def hist2d_demo(image):
    hsv = cv.cvtColor(image, cv.COLOR_BGR2HSV);
    # 得到一个直方图
    # hist = cv.calcHist(image, [0, 1], None, [180, 256], [0, 180, 0, 256])
    hist = cv.calcHist(image, [0, 1], None, [32, 32], [0, 180, 0, 256])
    print(hist.shape)
    plt.imshow(hist, interpolation='nearest')
    plt.title("2D Historgram")
    plt.show()


print("------------Hello Python------------")
src = cv.imread("C:/cvtest/0000.jpg")
# cv.namedWindow("input image", cv.WINDOW_AUTOSIZE)
# cv.imshow("input image", src)

# hist2d_demo(src)
back_projection_demo()

cv.waitKey(0)

cv.destroyAllWindows()