c++ - 使用 OpenCV c++ 进行对比度拉伸(stretch)的 RGB 彩色图像直方图

Question

我编写了一个程序，使用像素数据值制作灰度图像的直方图。我使用下面的代码段来增加图像访问像素值的对比度。我如何将这些逻辑应用于彩色图像以增加图像的对比度。gr 是灰度图像(从彩色图像转换而来)，gr_im 是新图像。

int x = 0;
    for (int  i = 0; i < h; i++)
    {
        for (int j = 0; j < w; j++) 
        {
            x = gr.at<uchar>(i,j);
            if (x < r1) 
            {
                gr_im.at<uchar>(i, j) = cvRound((s1/double(r1))*double(x));
            }
            else if(x <= r2){
                gr_im.at<uchar>(i, j) = cvRound((s2-s1 / double(r2-r1))*(x-r1) + s1);
            }
            else if (x > r2) {
                gr_im.at<uchar>(i, j) = cvRound((255-s2 / double(r1))*(x - r2) + s2);
            }
        }
    }

Answer 1

您可以使用类似下面的代码:

cv::Mat hsv;
cv::cvtColor(image, hsv, CV_BGR2HSV);
std::vector<cv::Mat> channels;
cv::split(hsv, channels)

// now call your function to specific channel

cv::merge(channels, hsv);
cv::cvtColor(hsv, image, CV_HSV2BGR);