c++ - 在 OpenCV Mat 和 Leptonica Pix 之间转换

Question

我需要在 C++ 中的 OpenCV Mat 图像和 Leptonica Pix 图像格式之间进行转换。这用于 8 位灰度图像的二值化。

Answer 1

我发现@ikarliga 的答案对我有用，因为我需要的是实际转换为 Mat 格式，不一定将它与 OP 所要求的 Tesseract API 一起使用。

这是我使用的两个功能。pix8ToMat 函数取自node-dv 项目

Pix *mat8ToPix(cv::Mat *mat8)
{
    Pix *pixd = pixCreate(mat8->size().width, mat8->size().height, 8);
    for(int y=0; y<mat8->rows; y++) {
        for(int x=0; x<mat8->cols; x++) {
            pixSetPixel(pixd, x, y, (l_uint32) mat8->at<uchar>(y,x));
        }
    }
    return pixd;
}

cv::Mat pix8ToMat(Pix *pix8)
{
    cv::Mat mat(cv::Size(pix8->w, pix8->h), CV_8UC1);
    uint32_t *line = pix8->data;
    for (uint32_t y = 0; y < pix8->h; ++y) {
        for (uint32_t x = 0; x < pix8->w; ++x) {
            mat.at<uchar>(y, x) = GET_DATA_BYTE(line, x);
        }
        line += pix8->wpl;
    }
    return mat;
}

Answer 2

我知道我迟到了大约 4 年，没有人在乎，但这项任务让我很沮丧，因为 leptonica 的文档太糟糕了。@loot
的答案的 问题是它不适用于彩色图像，所以这是我对他的（可能是粗略的）修改：pix8ToMat

cv::Mat pixToMat(Pix *pix) {
    int width = pixGetWidth(pix);
    int height = pixGetHeight(pix);
    int depth = pixGetDepth(pix);

    cv::Mat mat(cv::Size(width, height), depth == 1 ? CV_8UC1 : CV_8UC3);

    for (uint32_t y = 0; y < height; ++y) {
        for (uint32_t x = 0; x < width; ++x) {
            if (depth == 1) {
                l_uint32 val;
                pixGetPixel(pix, x, y, &val);
                mat.at<uchar>(cv::Point(x, y)) = static_cast<uchar>(255 * val);
            } else {
                l_int32 r, g, b;
                pixGetRGBPixel(pix, x, y, &r, &g, &b);

                cv::Vec3b color(b, g, r);
                mat.at<cv::Vec3b>(cv::Point(x, y)) = color;
            }
        }
    }

    return mat;
}

Answer 3

要将 pix 转换为 Mat，请在调用 pix8toMat 之前在 @loot 代码中添加这些行。

Pix *8bitPix = pixConvert1To8(NULL, pixt, 255, 0);

现在将这个 8bitPix 发送到 mat 转换。[它适用于二进制图像]