直方图反向投影

其他

发布时间 : 2021-01-02 15:22

阅读 :

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


def back_projection_demo():
    sample = cv.imread('./6.jpg')
    target = cv.imread('./7.jpg')
    roi_hsv = cv.cvtColor(sample, cv.COLOR_BGR2HSV)
    target_hsv = cv.cvtColor(target, cv.COLOR_BGR2HSV)

    cv.imshow('sample', sample)
    cv.imshow('target', target)

    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):
    hsv = cv.cvtColor(image, cv.COLOR_BGR2HSV)
    hist = cv.calcHist([image], [0, 1], None, [32, 32], [0, 180, 0, 256])
    # cv.imshow('hist2d', hist)
    plt.imshow(hist, interpolation='nearest')
    plt.title('2D Histogram')
    plt.show()


src = cv.imread('./0.png')  # blue, green, red
cv.namedWindow('input image', cv.WINDOW_AUTOSIZE)
# cv.imshow('input image', src)
# hist2d_demo(src)
back_projection_demo()
cv.waitKey(0)

cv.destroyAllWindows()

转载请注明来源，欢迎对文章中的引用来源进行考证，欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论，也可以邮件至 2621041184@qq.com