Skip to content

Commit

Permalink
Merge pull request tesseract-ocr#191 from amitdo/fix-184
Browse files Browse the repository at this point in the history
Fix tesseract-ocr#184. Training should work now
  • Loading branch information
zdenop committed Jan 17, 2016
2 parents d0a7679 + f6a1adb commit d94bd6c
Showing 1 changed file with 15 additions and 11 deletions.
26 changes: 15 additions & 11 deletions api/tesseractmain.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -286,7 +286,8 @@ void ParseArgs(const int argc, char** argv,
void PreloadRenderers(tesseract::TessBaseAPI* api,
tesseract::PointerVector<tesseract::TessResultRenderer>* renderers,
tesseract::PageSegMode pagesegmode,
const char* outputbase) {
const char* outputbase,
bool in_training_mode) {
if (pagesegmode == tesseract::PSM_OSD_ONLY) {
renderers->push_back(new tesseract::TessOsdRenderer(outputbase));
} else {
Expand Down Expand Up @@ -315,15 +316,8 @@ void PreloadRenderers(tesseract::TessBaseAPI* api,
renderers->push_back(new tesseract::TessBoxTextRenderer(outputbase));
}

// disable text renderer when using one of these configs:
// ambigs.train, box.train, box.train.stderr, linebox, rebox
bool disable_text_renderer =
(api->GetBoolVariable("tessedit_ambigs_training", &b) && b) ||
(api->GetBoolVariable("tessedit_resegment_from_boxes", &b) && b) ||
(api->GetBoolVariable("tessedit_make_boxes_from_boxes", &b) && b);

api->GetBoolVariable("tessedit_create_txt", &b);
if (b || (renderers->empty() && !disable_text_renderer)) {
if (b || (renderers->empty() && !in_training_mode)) {
renderers->push_back(new tesseract::TessTextRenderer(outputbase));
}
}
Expand Down Expand Up @@ -419,9 +413,19 @@ int main(int argc, char **argv) {
exit(ret_val);
}

// set in_training_mode to true when using one of these configs:
// ambigs.train, box.train, box.train.stderr, linebox, rebox
bool b = false;
bool in_training_mode =
(api.GetBoolVariable("tessedit_ambigs_training", &b) && b) ||
(api.GetBoolVariable("tessedit_resegment_from_boxes", &b) && b) ||
(api.GetBoolVariable("tessedit_make_boxes_from_boxes", &b) && b);

tesseract::PointerVector<tesseract::TessResultRenderer> renderers;
PreloadRenderers(&api, &renderers, pagesegmode, outputbase);
if (!renderers.empty()) {
PreloadRenderers(&api, &renderers, pagesegmode, outputbase,
in_training_mode);

if (!renderers.empty() || in_training_mode) {
bool succeed = api.ProcessPages(image, NULL, 0, renderers[0]);
if (!succeed) {
fprintf(stderr, "Error during processing.\n");
Expand Down

0 comments on commit d94bd6c

Please sign in to comment.