华拓科技网
您的当前位置:首页对一个文件夹内的图像进行直方图均衡化处理,自适应的直方图均衡化C++

对一个文件夹内的图像进行直方图均衡化处理,自适应的直方图均衡化C++

来源:华拓科技网
#include <iostream>
#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <filesystem>

using namespace cv;
using namespace std;

void adaptiveHistogramEqualization(const string& inputFolder, const string& outputFolder, int clipLimit, Size grid) {
    // 创建CLAHE(有限对比度自适应直方图均衡化器)
    Ptr<CLAHE> clahe = createCLAHE(clipLimit, grid);

    // 获取输入文件夹下的所有文件
    vector<String> filenames;
    glob(inputFolder + "/*.*", filenames); // 匹配所有文件类型

    for (size_t i = 0; i < filenames.size(); ++i) {
        // 读取图片
        Mat image = imread(filenames[i], IMREAD_GRAYSCALE);

        if (image.empty()) {
            cerr << "无法读取图像: " << filenames[i] << endl;
            continue;
        }

        // 对图像进行自适应直方图均衡化
        Mat equalizedImage;
        clahe->apply(image, equalizedImage);

        // 构造输出文件名
        string outputFileName = outputFolder + "/" + to_string(i) + "_equalized.jpg";

        // 保存结果图片
        imwrite(outputFileName, equalizedImage);
        cout << "已保存: " << outputFileName << endl;
    }
}

int main() {
    string inputFolder = ".\\input";   // 
    string outputFolder = ".\\output"; // 
    int clipLimit = 2; // 自适应直方图均衡化的clipLimit,可根据需要调整
    Size grid(8, 8);   // 自适应直方图均衡化的网格大小,可根据需要调整



    // 执行自适应直方图均衡化
    adaptiveHistogramEqualization(inputFolder, outputFolder, clipLimit, grid);

    return 0;
}

因篇幅问题不能全部显示,请点此查看更多更全内容