mirror/tesseract
1
0
Fork 0
mirror of https://github.com/tesseract-ocr/tesseract.git synced 2025-01-18 06:30:14 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

improve NO_CUBE_BUILD

Browse Source
This commit is contained in:
Zdenko Podobný 2015-08-09 18:09:52 +02:00
parent 67ede37b50
commit 545a0634da
2 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/baseapi.h
View File

@ -65,7 +65,9 @@ struct TBLOB;
namespace tesseract {
#ifndef NO_CUBE_BUILD
class CubeRecoContext;
#endif // NO_CUBE_BUILD
class Dawg;
class Dict;
class EquationDetect;
@ -735,8 +737,10 @@ class TESS_API TessBaseAPI {
void InitTruthCallback(TruthCallback *cb) { truth_cb_ = cb; }
#ifndef NO_CUBE_BUILD
/** Return a pointer to underlying CubeRecoContext object if present. */
CubeRecoContext *GetCubeRecoContext() const;
#endif // NO_CUBE_BUILD
void set_min_orientation_margin(double margin);

23
training/classifier_tester.cpp
View File

@ -19,10 +19,12 @@
#include <stdio.h>
#ifndef USE_STD_NAMESPACE
#include "base/commandlineflags.h"
#endif
#endif // USE_STD_NAMESPACE
#include "baseapi.h"
#include "commontraining.h"
#ifndef NO_CUBE_BUILD
#include "cubeclassifier.h"
#endif // NO_CUBE_BUILD
#include "mastertrainer.h"
#include "params.h"
#include "strngs.h"
@ -37,12 +39,18 @@ DECLARE_STRING_PARAM_FLAG(T);
enum ClassifierName {
CN_PRUNER,
CN_FULL,
#ifndef NO_CUBE_BUILD
CN_CUBE,
CN_CUBETESS,
#endif // NO_CUBE_BUILD
CN_COUNT
};
const char* names[] = {"pruner", "full", "cube", "cubetess", NULL };
const char* names[] = {"pruner", "full",
#ifndef NO_CUBE_BUILD
"cube", "cubetess",
#endif // NO_CUBE_BUILD
NULL };
static tesseract::ShapeClassifier* InitializeClassifier(
const char* classifer_name, const UNICHARSET& unicharset,
@ -64,13 +72,20 @@ static tesseract::ShapeClassifier* InitializeClassifier(
// We need to initialize tesseract to test.
*api = new tesseract::TessBaseAPI;
tesseract::OcrEngineMode engine_mode = tesseract::OEM_TESSERACT_ONLY;
#ifndef NO_CUBE_BUILD
if (classifier == CN_CUBE || classifier == CN_CUBETESS)
engine_mode = tesseract::OEM_TESSERACT_CUBE_COMBINED;
#endif // NO_CUBE_BUILD
tesseract::Tesseract* tesseract = NULL;
tesseract::Classify* classify = NULL;
if (classifier == CN_CUBE || classifier == CN_CUBETESS ||
if (
#ifndef NO_CUBE_BUILD
classifier == CN_CUBE || classifier == CN_CUBETESS ||
#endif // NO_CUBE_BUILD
classifier == CN_PRUNER || classifier == CN_FULL) {
#ifndef NO_CUBE_BUILD
(*api)->SetVariable("cube_debug_level", "2");
#endif // NO_CUBE_BUILD
if ((*api)->Init(FLAGS_tessdata_dir.c_str(), FLAGS_lang.c_str(),
engine_mode) < 0) {
fprintf(stderr, "Tesseract initialization failed!\n");
@ -96,10 +111,12 @@ static tesseract::ShapeClassifier* InitializeClassifier(
shape_classifier = new tesseract::TessClassifier(true, classify);
} else if (classifier == CN_FULL) {
shape_classifier = new tesseract::TessClassifier(false, classify);
#ifndef NO_CUBE_BUILD
} else if (classifier == CN_CUBE) {
shape_classifier = new tesseract::CubeClassifier(tesseract);
} else if (classifier == CN_CUBETESS) {
shape_classifier = new tesseract::CubeTessClassifier(tesseract);
#endif // NO_CUBE_BUILD
} else {
fprintf(stderr, "%s tester not yet implemented\n", classifer_name);
return NULL;