OpenCV 将rgb图像转化成字符图像

时间:2024-03-12 12:38:24

将RGB图像转换成字符图像(ASCII art)通常涉及到灰度化、降采样、映射字符等一系列步骤。以下是一个简化的OpenCV+C++实现示例:

#include <opencv2/opencv.hpp>
#include <iostream>
#include <string>

// 字符映射表,从亮到暗
std::string charset = " .:-=+*#%@";

// 将RGB图像转为字符图像
std::string RGBToASCII(const cv::Mat& img) {
    // 1. 转换为灰度图像
    cv::Mat gray;
    cv::cvtColor(img, gray, cv::COLOR_BGR2GRAY);

    // 2. 降采样,缩小图像以便适应字符分辨率
    cv::resize(gray, gray, cv::Size(40, 40), 0.1, 0.1);  // 此处0.1只是一个示例,具体比例需根据终端窗口大小和字符数量调整

    // 3. 将灰度值映射为字符索引
    std::string asciiArt;
    for (int i = 0; i < gray.rows; ++i) {
        for (int j = 0; j < gray.cols; ++j) {
            // 获取当前像素的灰度值,并将其归一化到charset长度范围内
            double val = gray.at<uchar>(i, j) / 255.0 * (charset.size() - 1);
            size_t index = static_cast<size_t>(val);

            // 添加对应字符到字符串
            asciiArt += charset[index];
                              asciArt += ' ';
        }
        asciiArt += '\n'; // 添加换行符
    }

    return asciiArt;
}

int main() {
    cv::Mat img = cv::imread("input.jpg", cv::IMREAD_COLOR);
    if (img.empty()) {
        std::cerr << "Could not open or find the image!" << std::endl;
        return -1;
    }

    std::string asciiArt = RGBToASCII(img);

    // 输出或保存字符图像
    std::cout << asciiArt << std::endl;

    // 或者保存到文件
    std::ofstream file("ascii_art.txt");
    if (file.is_open()) {
        file << asciiArt;
        file.close();
    } else {
        std::cerr << "Unable to open file for writing." << std::endl;
    }

    return 0;
}

这段代码首先将RGB图像转换为灰度图像,然后对其进行降采样,最后将每个像素的灰度值映射到字符集中的字符,形成ASCII艺术图像。这里的字符集可以根据需要自定义,亮的像素映射到字符集中靠前的字符,暗的像素映射到靠后的字符。此外,降采样的比例应根据输出设备的实际分辨率进行调整。