转换cv :: Mat for tesseract

时间:2011-11-13 22:44:43

标签: c++ opencv tesseract

我正在使用OpenCV来提取扫描文档的子图像,并希望使用tesseract对此子图像执行OCR。

我发现我可以在tesseract中使用两种方法进行文本识别,但到目前为止我还没有找到可行的解决方案。

答。)如何将cv::Mat转换为PIX*  (PIX *是leptonica

的数据类型

根据下面的vasiles代码,这基本上是我当前的代码:

 cv::Mat image = cv::imread("c:/image.png"); 
 cv::Mat subImage = image(cv::Rect(50, 200, 300, 100)); 

 int depth;
 if(subImage.depth() == CV_8U)
    depth = 8;
 //other cases not considered yet

 PIX* pix = pixCreateHeader(subImage.size().width, subImage.size().height, depth);
 pix->data = (l_uint32*) subImage.data; 

 tesseract::TessBaseAPI tess;
 STRING text; 
 if(tess.ProcessPage(pix, 0, 0, &text))
 {
    std::cout << text.string(); 
 }   

虽然它没有崩溃或任何事情,但OCR结果仍然是错误的。它应该识别我的示例图像中的一个单词,但它会返回一些不可读的字符。

方法PIX_HEADER不存在,因此我使用pixCreateHeader,但它不会将通道数作为参数。那么如何设置通道数呢?

B。)如何cv::Mat使用TesseractRect()

Tesseract提供了另一种使用此签名进行文本识别的方法:

char * TessBaseAPI::TesseractRect   (   
    const UINT8 *   imagedata,
    int     bytes_per_pixel,
    int     bytes_per_line,
    int     left,
    int     top,
    int     width,
    int     height   
)   

目前我使用的是以下代码,但它也会返回不可读的字符(虽然不同于上面的代码。

char* cr = tess.TesseractRect(
           subImage.data, 
           subImage.channels(), 
           subImage.channels() * subImage.size().width, 
           0, 
           0, 
           subImage.size().width, 
           subImage.size().height);   

4 个答案:

答案 0 :(得分:15)

tesseract::TessBaseAPI tess; 
cv::Mat sub = image(cv::Rect(50, 200, 300, 100));
tess.SetImage((uchar*)sub.data, sub.size().width, sub.size().height, sub.channels(), sub.step1());
tess.Recognize(0);
const char* out = tess.GetUTF8Text();

答案 1 :(得分:4)

对于任何使用OpenCV / Tesseract的JavaCPP预设的人来说,这是有效的

Mat img = imread("file.jpg");
Mat gray = new Mat();
cvtColor(img, gray, CV_BGR2GRAY);

// api is a Tesseract client which is initialised

api.SetImage(gray.data().asBuffer(),gray.size().width(),gray.size().height(),gray.channels(),gray.size1())

答案 2 :(得分:1)

cv::Mat image = cv::imread(argv[1]);

cv::Mat gray;
cv::cvtColor(image, gray, CV_BGR2GRAY);

PIX *pixS = pixCreate(gray.size().width, gray.size().height, 8);

for(int i=0; i<gray.rows; i++) 
    for(int j=0; j<gray.cols; j++) 
        pixSetPixel(pixS, j,i, (l_uint32) gray.at<uchar>(i,j));

答案 3 :(得分:0)

首先,制作subImage的深层副本,以便将其存储在一个连续的内存块中:

cv::Mat subImage = image(cv::Rect(50, 200, 300, 100)).clone(); 

然后,使用正确的参数初始化一个PIX(我不知道如何)。

// ???? Put your own constructor here. 
PIX* pix = new PIX_HEADER(width, height, channels, depth); 

或者,手动创建:

PIX pix;
pix.width = subImage.width;
...

然后将pix数据指针设置为subImage数据指针

pix.data = subImage.data;

最后,在使用pix完成工作之前,请确保您的subImage对象不会超出范围。