Merge pull request #45 from lluisgomez/master

Adds OCRTesseract class and sample demo

modules/text/CMakeLists.txt

+set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR})
+
+find_package(Tesseract)
+if(Tesseract_FOUND)
+    message(STATUS "Tesseract:   YES")
+    set(HAVE_TESSERACT 1)
+else()
+    message(STATUS "Tesseract:   NO")
+endif()
+
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/text_config.hpp.in
+               ${CMAKE_BINARY_DIR}/text_config.hpp @ONLY)
+
+include_directories(${CMAKE_CURRENT_BINARY_DIR})
+
+if(${Tesseract_FOUND})
+include_directories(${Tesseract_INCLUDE_DIR})
+endif()
+
 set(the_description "Text Detection and Recognition")
 ocv_define_module(text opencv_ml opencv_highgui opencv_imgproc opencv_core)
+
+if(${Tesseract_FOUND})
+  target_link_libraries(opencv_text ${Tesseract_LIBS})
+endif()

modules/text/FindTesseract.cmake

+# Tesseract OCR
+unset(Tesseract_FOUND)
+
+find_path(Tesseract_INCLUDE_DIR tesseract/baseapi.h
+  HINTS
+  /usr/include
+  /usr/local/include)
+
+find_library(Tesseract_LIBRARY NAMES tesseract
+  HINTS
+  /usr/lib
+  /usr/local/lib)
+
+find_library(Lept_LIBRARY NAMES lept
+  HINTS
+  /usr/lib
+  /usr/local/lib)
+
+set(Tesseract_LIBS ${Tesseract_LIBRARY} ${Lept_LIBRARY})
+if(Tesseract_LIBS AND Tesseract_INCLUDE_DIR)
+    set(Tesseract_FOUND 1)
+endif()
+
+        

modules/text/include/opencv2/text.hpp

@@ -40,5 +40,6 @@ the use of this software, even if advised of the possibility of such damage.
 #define __OPENCV_TEXT_HPP__
 
 #include "opencv2/text/erfilter.hpp"
+#include "opencv2/text/ocr.hpp"
 
 #endif

modules/text/include/opencv2/text/ocr.hpp

+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+//  By downloading, copying, installing or using the software you agree to this license.
+//  If you do not agree to this license, do not download, install,
+//  copy or use the software.
+//
+//
+//                          License Agreement
+//                For Open Source Computer Vision Library
+//
+// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
+// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
+// Copyright (C) 2013, OpenCV Foundation, all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//   * Redistribution's of source code must retain the above copyright notice,
+//     this list of conditions and the following disclaimer.
+//
+//   * Redistribution's in binary form must reproduce the above copyright notice,
+//     this list of conditions and the following disclaimer in the documentation
+//     and/or other materials provided with the distribution.
+//
+//   * The name of the copyright holders may not be used to endorse or promote products
+//     derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Intel Corporation or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//
+//M*/
+
+#ifndef __OPENCV_TEXT_OCR_HPP__
+#define __OPENCV_TEXT_OCR_HPP__
+
+#include "text_config.hpp"
+
+#ifdef HAVE_TESSERACT
+#include <tesseract/baseapi.h>
+#include <tesseract/resultiterator.h>
+#endif
+
+#include "opencv2/core.hpp"
+#include <vector>
+#include <string>
+
+
+namespace cv
+{
+namespace text
+{
+
+using namespace std;
+
+enum
+{
+    OCR_LEVEL_WORD,
+    OCR_LEVEL_TEXTLINE
+};
+
+#ifdef HAVE_TESSERACT
+class CV_EXPORTS OCRTesseract
+{
+private:
+    tesseract::TessBaseAPI tess;
+
+public:
+    //Default constructor
+    OCRTesseract(const char* datapath=NULL, const char* language=NULL, const char* char_whitelist=NULL,
+                 tesseract::OcrEngineMode oem=tesseract::OEM_DEFAULT, tesseract::PageSegMode psmode=tesseract::PSM_AUTO);
+
+    ~OCRTesseract();
+
+    void run(Mat& image, string& output_text, vector<Rect>* component_rects=NULL,
+             vector<string>* component_texts=NULL, vector<float>* component_confidences=NULL,
+             int component_level=0);
+};
+#else
+//stub
+class CV_EXPORTS OCRTesseract
+{
+public:
+    //Default constructor
+    OCRTesseract(const char* datapath=NULL, const char* language=NULL, const char* char_whitelist=NULL,
+                 int oem=0, int psmode=0);
+
+    ~OCRTesseract();
+
+    void run(Mat& image, string& output_text, vector<Rect>* component_rects=NULL,
+             vector<string>* component_texts=NULL, vector<float>* component_confidences=NULL,
+             int component_level=0);
+};
+#endif
+
+
+
+}
+}
+#endif // _OPENCV_TEXT_OCR_HPP_

modules/text/src/ocr.cpp

+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+//  By downloading, copying, installing or using the software you agree to this license.
+//  If you do not agree to this license, do not download, install,
+//  copy or use the software.
+//
+//
+//                           License Agreement
+//                For Open Source Computer Vision Library
+//
+// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
+// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//   * Redistribution's of source code must retain the above copyright notice,
+//     this list of conditions and the following disclaimer.
+//
+//   * Redistribution's in binary form must reproduce the above copyright notice,
+//     this list of conditions and the following disclaimer in the documentation
+//     and/or other materials provided with the distribution.
+//
+//   * The name of the copyright holders may not be used to endorse or promote products
+//     derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Intel Corporation or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//
+//M*/
+
+#include "precomp.hpp"
+#include "opencv2/imgproc.hpp"
+#include "opencv2/ml.hpp"
+
+#include <iostream>
+#include <fstream>
+#include <queue>
+
+namespace cv
+{
+namespace text
+{
+
+using namespace std;
+
+
+#ifdef HAVE_TESSERACT
+//Default constructor
+OCRTesseract::OCRTesseract(const char* datapath, const char* language, const char* char_whitelist, tesseract::OcrEngineMode oemode, tesseract::PageSegMode psmode)
+{
+
+    const char *lang = "eng";
+    if (language != NULL)
+        lang = language;
+
+    if (tess.Init(datapath, lang, oemode))
+    {
+        cout << "OCRTesseract: Could not initialize tesseract." << endl;
+        throw 1;
+    }
+
+    //cout << "OCRTesseract: tesseract version " << tess.Version() << endl;
+
+    tesseract::PageSegMode pagesegmode = psmode;
+    tess.SetPageSegMode(pagesegmode);
+
+    if(char_whitelist != NULL)
+        tess.SetVariable("tessedit_char_whitelist", char_whitelist);
+    else
+        tess.SetVariable("tessedit_char_whitelist", "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ");
+
+    tess.SetVariable("save_best_choices", "T");
+
+}
+
+OCRTesseract::~OCRTesseract()
+{
+    tess.End();
+}
+
+void OCRTesseract::run(Mat& image, string& output, vector<Rect>* component_rects,
+                       vector<string>* component_texts, vector<float>* component_confidences, int component_level)
+{
+    CV_Assert( (image.type() == CV_8UC1) || (image.type() == CV_8UC1) );
+    if (component_texts != 0)
+        component_texts->clear();
+    if (component_rects != 0)
+        component_rects->clear();
+    if (component_confidences != 0)
+        component_confidences->clear();
+
+    tess.SetImage((uchar*)image.data, image.size().width, image.size().height, image.channels(), image.step1());
+    tess.Recognize(0);
+    output = string(tess.GetUTF8Text());
+
+    if ( (component_rects != NULL) || (component_texts != NULL) || (component_confidences != NULL) )
+    {
+        tesseract::ResultIterator* ri = tess.GetIterator();
+        tesseract::PageIteratorLevel level = tesseract::RIL_WORD;
+        if (component_level == OCR_LEVEL_TEXTLINE)
+            level = tesseract::RIL_TEXTLINE;
+
+        if (ri != 0) {
+            do {
+                const char* word = ri->GetUTF8Text(level);
+                if (word == NULL)
+                    continue;
+                float conf = ri->Confidence(level);
+                int x1, y1, x2, y2;
+                ri->BoundingBox(level, &x1, &y1, &x2, &y2);
+
+                if (component_texts != 0)
+                    component_texts->push_back(string(word));
+                if (component_rects != 0)
+                    component_rects->push_back(Rect(x1,y1,x2-x1,y2-y1));
+                if (component_confidences != 0)
+                    component_confidences->push_back(conf);
+
+                delete[] word;
+            } while (ri->Next(level));
+        }
+        delete ri;
+    }
+
+    tess.Clear();
+}
+#else
+//Stub constructor
+OCRTesseract::OCRTesseract(const char* datapath, const char* language, const char* char_whitelist, int oemode, int psmode)
+{
+    cout << "OCRTesseract("<<oemode<<psmode<<"): Tesseract not found." << endl;
+    if (datapath != NULL)
+      cout << "            " << datapath << endl;
+    if (language != NULL)
+      cout << "            " << language << endl;
+    if (char_whitelist != NULL)
+      cout << "            " << char_whitelist << endl;
+}
+
+//Stub destructor
+OCRTesseract::~OCRTesseract()
+{
+}
+
+//Stub method, does nothing
+void OCRTesseract::run(Mat& image, string& output, vector<Rect>* component_rects,
+                       vector<string>* component_texts, vector<float>* component_confidences, int component_level)
+{
+    CV_Assert( (image.type() == CV_8UC1) || (image.type() == CV_8UC1) );
+
+    cout << "OCRTesseract(" << component_level << image.type() <<"): Tesseract not found." << endl;
+    output.clear();
+    if(component_rects)
+        component_rects->clear();
+    if(component_texts)
+        component_texts->clear();
+    if(component_confidences)
+        component_confidences->clear();
+}
+#endif
+
+
+
+}
+}

modules/text/text_config.hpp.in

+#ifndef __OPENCV_TEXT_CONFIG_HPP__
+#define __OPENCV_TEXT_CONFIG_HPP__
+
+// HAVE OCR Tesseract
+#cmakedefine HAVE_TESSERACT
+
+#endif
\ No newline at end of file