Python opencv学习-13 直方图反向投影（用于图像分割）

代码和图片大量参考https://blog.csdn.net/tengfei461807914/article/details/77075567

自己跑了下验证了下~

个人理解： 直方图反向投影用来做图像分割，或者说知道了一个目标的图片的一部分，去另一个包含这个目标的图中去分割这个目标，如去图2中找图1 的部分

图1

图2

import cv2
import numpy as np
from matplotlib import pyplot as plt
#roi图片，就想要找的的图片
roi = cv2.imread('image1/small.png')
hsv = cv2.cvtColor(roi,cv2.COLOR_BGR2HSV)

#目标搜索图片
target = cv2.imread('image1/big.png')
hsvt = cv2.cvtColor(target,cv2.COLOR_BGR2HSV)

#计算目标直方图
roihist = cv2.calcHist([hsv],[0,1],None,[180,256],[0,180,0,256])

#归一化，参数为原图像和输出图像，归一化后值全部在2到255范围
cv2.normalize(roihist,roihist,0,255,cv2.NORM_MINMAX)
# 实现反向投影hsvt为目标搜索图片， roihist为要找图片的直方图
dst = cv2.calcBackProject([hsvt],[0,1],roihist,[0,180,0,256],1)

#卷积连接分散的点
disc = cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(5,5))
dst = cv2.filter2D(dst,-1,disc)

# 将获取到的概率图像二值化，获取到掩膜
ret,thresh = cv2.threshold(dst,50,255,0)
#使用merge变成3通道图像（因为原图是3通道）
thresh = cv2.merge((thresh,thresh,thresh))

#蒙板
res = cv2.bitwise_and(target,thresh)
#矩阵按列合并,就是把target,thresh和res三个图片横着拼在一起
res = np.hstack((target,thresh,res))

cv2.imwrite('res.jpg',res)
#显示图像
cv2.imshow('1',res)
cv2.waitKey(0)