在OpenCV Mat和Leptonica Pix之间转换

Convert between OpenCV Mat and Leptonica Pix

本文关键字:Pix 之间 转换 Leptonica OpenCV Mat      更新时间:2023-10-16

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

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

下面是我使用的两个函数。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;
}

我知道我已经迟了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;
}

要将pix转换为Mat,请在调用pix8toMat之前在@loot代码中添加以下行。

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

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