python 重写cv2.calcHist

如果你想重写 `cv2.calcHist()` 函数，你可以参考以下代码：

import numpy as np

def calc_hist(image, channels=[0], mask=None, histSize=[256], ranges=[0, 256]):
    # 创建一个空的直方图
    hist = np.zeros(histSize, dtype=np.float32)
    
    # 计算直方图
    if mask is None:
        # 如果没有掩膜，直接计算
        hist = cv2.calcHist([image], channels, None, histSize, ranges)
    else:
        # 如果有掩膜，先将掩膜转换为非零值，再计算直方图
        mask = mask.astype(np.uint8)
        image = cv2.bitwise_and(image, image, mask=mask)
        hist = cv2.calcHist([image], channels, None, histSize, ranges)
    
    return hist

这个函数与 `cv2.calcHist()` 函数的参数基本相同，但它返回的是一个 `numpy` 数组而不是一个 OpenCV 直方图对象。此外，它还添加了一个名为 `mask` 的参数，以便可以在计算直方图时使用掩膜。

相关问题

cv2.calcHist python重写

cv2.calcHist是OpenCV库中用于计算图像直方图的函数。如果需要在Python中重写该函数，可以使用numpy库中的histogram函数来实现。以下是一个简单的示例代码：

import numpy as np

def calcHist(img, bins, ranges):
    # 将图像转换为灰度图像
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    # 计算直方图
    hist, _ = np.histogram(gray, bins, ranges)
    # 返回直方图
    return hist

该函数接受三个参数：图像、直方图的bin数和像素值范围。它首先将图像转换为灰度图像，然后计算直方图，并返回一个numpy数组表示直方图。注意，此处只返回直方图而不返回bin中心。如果需要返回bin中心，请使用numpy库中的histogram函数的第二个返回值。

cv2.calcHist实现

cv2.calcHist是OpenCV中的函数，用于计算图像的直方图。下面是一个简单的示例，演示如何使用cv2.calcHist计算图像的灰度直方图：

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

img = cv2.imread('lena.jpg', 0)

hist = cv2.calcHist([img], [0], None, [256], [0, 256])

plt.plot(hist)
plt.xlim([0, 256])
plt.show()

在上面的代码中，我们首先使用cv2.imread函数读取一张灰度图像，并将其存储在img变量中。然后，我们使用cv2.calcHist函数计算img的灰度直方图，将结果存储在hist变量中。在这里，我们使用了单通道图像（[0]）和256个bin。最后，我们使用matplotlib库绘制直方图。

如果您想计算彩色图像的直方图，可以使用以下代码：

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

img = cv2.imread('lena.jpg')

color = ('b','g','r')

for i,col in enumerate(color):
    hist = cv2.calcHist([img],[i],None,[256],[0,256])
    plt.plot(hist,color = col)
    plt.xlim([0,256])

plt.show()

在这里，我们首先使用cv2.imread函数读取一张彩色图像，并将其存储在img变量中。然后，我们使用三通道图像（[i]）和256个bin计算每个通道的直方图。最后，我们使用matplotlib库将三个直方图绘制在同一个图像上。