Skip to content

Commit 5c89c78

Browse files
committed
Adds OCRTesseract class and sample demo
1 parent 70a2bca commit 5c89c78

File tree

7 files changed

+685
-0
lines changed

7 files changed

+685
-0
lines changed

modules/text/CMakeLists.txt

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,25 @@
1+
set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR})
2+
3+
find_package(Tesseract)
4+
if(Tesseract_FOUND)
5+
message(STATUS "Tesseract: YES")
6+
set(HAVE_TESSERACT 1)
7+
else()
8+
message(STATUS "Tesseract: NO")
9+
endif()
10+
11+
configure_file(${CMAKE_CURRENT_SOURCE_DIR}/text_config.hpp.in
12+
${CMAKE_BINARY_DIR}/text_config.hpp @ONLY)
13+
14+
include_directories(${CMAKE_CURRENT_BINARY_DIR})
15+
16+
if(${Tesseract_FOUND})
17+
include_directories(${Tesseract_INCLUDE_DIR})
18+
endif()
19+
120
set(the_description "Text Detection and Recognition")
221
ocv_define_module(text opencv_ml opencv_highgui opencv_imgproc opencv_core)
22+
23+
if(${Tesseract_FOUND})
24+
target_link_libraries(opencv_text ${Tesseract_LIBS})
25+
endif()

modules/text/FindTesseract.cmake

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# Tesseract OCR
2+
unset(Tesseract_FOUND)
3+
4+
find_path(Tesseract_INCLUDE_DIR tesseract/baseapi.h
5+
HINTS
6+
/usr/include
7+
/usr/local/include)
8+
9+
find_library(Tesseract_LIBRARY NAMES tesseract
10+
HINTS
11+
/usr/lib
12+
/usr/local/lib)
13+
14+
find_library(Lept_LIBRARY NAMES lept
15+
HINTS
16+
/usr/lib
17+
/usr/local/lib)
18+
19+
set(Tesseract_LIBS ${Tesseract_LIBRARY} ${Lept_LIBRARY})
20+
if(Tesseract_LIBS AND Tesseract_INCLUDE_DIR)
21+
set(Tesseract_FOUND 1)
22+
endif()
23+
24+

modules/text/include/opencv2/text.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,5 +40,6 @@ the use of this software, even if advised of the possibility of such damage.
4040
#define __OPENCV_TEXT_HPP__
4141

4242
#include "opencv2/text/erfilter.hpp"
43+
#include "opencv2/text/ocr.hpp"
4344

4445
#endif
Lines changed: 110 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,110 @@
1+
/*M///////////////////////////////////////////////////////////////////////////////////////
2+
//
3+
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
4+
//
5+
// By downloading, copying, installing or using the software you agree to this license.
6+
// If you do not agree to this license, do not download, install,
7+
// copy or use the software.
8+
//
9+
//
10+
// License Agreement
11+
// For Open Source Computer Vision Library
12+
//
13+
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
14+
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
15+
// Copyright (C) 2013, OpenCV Foundation, all rights reserved.
16+
// Third party copyrights are property of their respective owners.
17+
//
18+
// Redistribution and use in source and binary forms, with or without modification,
19+
// are permitted provided that the following conditions are met:
20+
//
21+
// * Redistribution's of source code must retain the above copyright notice,
22+
// this list of conditions and the following disclaimer.
23+
//
24+
// * Redistribution's in binary form must reproduce the above copyright notice,
25+
// this list of conditions and the following disclaimer in the documentation
26+
// and/or other materials provided with the distribution.
27+
//
28+
// * The name of the copyright holders may not be used to endorse or promote products
29+
// derived from this software without specific prior written permission.
30+
//
31+
// This software is provided by the copyright holders and contributors "as is" and
32+
// any express or implied warranties, including, but not limited to, the implied
33+
// warranties of merchantability and fitness for a particular purpose are disclaimed.
34+
// In no event shall the Intel Corporation or contributors be liable for any direct,
35+
// indirect, incidental, special, exemplary, or consequential damages
36+
// (including, but not limited to, procurement of substitute goods or services;
37+
// loss of use, data, or profits; or business interruption) however caused
38+
// and on any theory of liability, whether in contract, strict liability,
39+
// or tort (including negligence or otherwise) arising in any way out of
40+
// the use of this software, even if advised of the possibility of such damage.
41+
//
42+
//M*/
43+
44+
#ifndef __OPENCV_TEXT_OCR_HPP__
45+
#define __OPENCV_TEXT_OCR_HPP__
46+
47+
#include "text_config.hpp"
48+
49+
#ifdef HAVE_TESSERACT
50+
#include <tesseract/baseapi.h>
51+
#include <tesseract/resultiterator.h>
52+
#endif
53+
54+
#include "opencv2/core.hpp"
55+
#include <vector>
56+
#include <string>
57+
58+
59+
namespace cv
60+
{
61+
namespace text
62+
{
63+
64+
using namespace std;
65+
66+
enum
67+
{
68+
OCR_LEVEL_WORD,
69+
OCR_LEVEL_TEXTLINE
70+
};
71+
72+
#ifdef HAVE_TESSERACT
73+
class CV_EXPORTS OCRTesseract
74+
{
75+
private:
76+
tesseract::TessBaseAPI tess;
77+
78+
public:
79+
//Default constructor
80+
OCRTesseract(const char* datapath=NULL, const char* language=NULL, const char* char_whitelist=NULL,
81+
tesseract::OcrEngineMode oem=tesseract::OEM_DEFAULT, tesseract::PageSegMode psmode=tesseract::PSM_AUTO);
82+
83+
~OCRTesseract();
84+
85+
void run(Mat& image, string& output_text, vector<Rect>* component_rects=NULL,
86+
vector<string>* component_texts=NULL, vector<float>* component_confidences=NULL,
87+
int component_level=0);
88+
};
89+
#else
90+
//stub
91+
class CV_EXPORTS OCRTesseract
92+
{
93+
public:
94+
//Default constructor
95+
OCRTesseract(const char* datapath=NULL, const char* language=NULL, const char* char_whitelist=NULL,
96+
int oem=0, int psmode=0);
97+
98+
~OCRTesseract();
99+
100+
void run(Mat& image, string& output_text, vector<Rect>* component_rects=NULL,
101+
vector<string>* component_texts=NULL, vector<float>* component_confidences=NULL,
102+
int component_level=0);
103+
};
104+
#endif
105+
106+
107+
108+
}
109+
}
110+
#endif // _OPENCV_TEXT_OCR_HPP_

0 commit comments

Comments
 (0)