OpenCV Python equalizeHist 彩色图像 [英] OpenCV Python equalizeHist colored image

问题描述

我需要对彩色图像进行直方图均衡.

I need to do a histogram equalization for a colored image.

首先我将彩色图像转换为灰色并将其提供给 equalizeHist 函数:

First I convert the colored image to gray and give it to the equalizeHist function:

image = cv2.imread("photo.jpg")
image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
cv2.equalizeHist(image)
cv2.imshow("equalizeHist", image)
cv2.waitKey(0)

但在此之后我需要将图像转换回 RGB；我该怎么做?

But after this I need to convert the image back to RGB; how can i do that?

推荐答案

来源:https://www.packtpub.com/packtlib/book/Application-Development/9781785283932/2/ch02lvl1sec26/Enhancing%20the%20contrast%20in%2020图像

import cv2
import numpy as np

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

img_yuv = cv2.cvtColor(img, cv2.COLOR_BGR2YUV)

# equalize the histogram of the Y channel
img_yuv[:,:,0] = cv2.equalizeHist(img_yuv[:,:,0])

# convert the YUV image back to RGB format
img_output = cv2.cvtColor(img_yuv, cv2.COLOR_YUV2BGR)

cv2.imshow('Color input image', img)
cv2.imshow('Histogram equalized', img_output)

cv2.waitKey(0)

~原始链接不再可用，这里实现了类似的想法:使用 OpenCV 对彩色图像进行直方图均衡化

~edit: original link is no longer available, similar idea is implemented here: Histogram Equalization of a Color image with OpenCV

这篇关于OpenCV Python equalizeHist 彩色图像的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！