Question

我知道可以均衡图像的直方图，如：

equalizeHist(image, image);

如果我想要我可以定义一个roi并在图像中均衡这个roi：

Mat aux3 = image.clone();
equalizeHist(image(Rect(0,100, 200,200)), aux3(Rect(0,100, 200,200)));

我现在想做什么（我不知道是否可能），是使用点矢量（cv ::矢量轮廓）定义一个Roi（轮廓）并均衡这个roi（这个roi）并不总是一个矩形）

所以，这里的问题是：

是否可以使用openCV函数均衡图像的非矩形部分？

Answer 1

OpenCV中没有内置函数可以使用掩码执行直方图均衡。你仍然可以写自定义的。

获取灰度图像：

// Load image
Mat3b img = imread("path_to_image");

// Convert to grayscale
Mat1b gray;
cvtColor(img, gray, COLOR_BGR2GRAY);

从您的观点中形成一个面具：

// Your vector of points
vector<Point> pts = { Point(300, 180), Point(450, 150), Point(600, 200), Point(650, 350), Point(300,300) };

// Create the mask
Mat1b mask(img.rows, img.cols, uchar(0));
vector<vector<Point>> ptsarray{pts};
fillPoly(mask, ptsarray, Scalar(255));

调用自定义函数equalizeHistWithMask，使用掩码均衡图像：

// Equalize with mask
Mat1b equalized;
equalizeHistWithMask(gray, equalized, mask);

此处为完整的参考代码，包含equalizeHistWithMask函数：

#include <iostream>
#include <vector>
#include <algorithm>
#include <opencv2/opencv.hpp>
using namespace std;
using namespace cv;

void equalizeHistWithMask(const Mat1b& src, Mat1b& dst, Mat1b mask = Mat1b())
{
    int cnz = countNonZero(mask);
    if (mask.empty() || ( cnz == src.rows*src.cols))
    {
        equalizeHist(src, dst);
        return;
    }

    dst = src.clone();

    // Histogram
    vector<int> hist(256,0);
    for (int r = 0; r < src.rows; ++r) {
        for (int c = 0; c < src.cols; ++c) {
            if (mask(r, c)) {
                hist[src(r, c)]++;
            }
        }
    }

    // Cumulative histogram
    float scale = 255.f / float(cnz);
    vector<uchar> lut(256);
    int sum = 0;
    for (int i = 0; i < hist.size(); ++i) {
        sum += hist[i];
        lut[i] = saturate_cast<uchar>(sum * scale);
    }

    // Apply equalization
    for (int r = 0; r < src.rows; ++r) {
        for (int c = 0; c < src.cols; ++c) {
            if (mask(r, c)) {
                dst(r, c) = lut[src(r,c)];
            }
        }
    }
}

int main()
{
    // Load image
    Mat3b img = imread("path_to_image");

    // Convert to grayscale
    Mat1b gray;
    cvtColor(img, gray, COLOR_BGR2GRAY);

    // Your vector of points
    vector<Point> pts = { Point(300, 180), Point(450, 150), Point(600, 200), Point(650, 350), Point(300,300) };

    // Create the mask
    Mat1b mask(img.rows, img.cols, uchar(0));
    vector<vector<Point>> ptsarray{pts};
    fillPoly(mask, ptsarray, Scalar(255));

    // Equalize with mask
    Mat1b equalized;
    equalizeHistWithMask(gray, equalized, mask);

    imshow("Gray", gray);
    imshow("Mask", mask);
    imshow("Equalized", equalized);
    waitKey();

    return 0;
}

<强>积分

该代码基于 answers.opencv.org上的this问题

OpenCV，均衡图像轮廓的直方图

1 个答案: