classifier_tester.cpp File Reference

#include <tesseract/baseapi.h>
#include <algorithm>
#include <cstdio>
#include "commontraining.h"
#include "mastertrainer.h"
#include "params.h"
#include "tessclassifier.h"
#include "tesseractclass.h"

Go to the source code of this file.

Enumerations
enum	ClassifierName { CN_PRUNER , CN_FULL , CN_COUNT }

Functions
int	main (int argc, char **argv)

Enumeration Type Documentation

ClassifierName

enum ClassifierName

Enumerator
CN_PRUNER
CN_FULL
CN_COUNT

Definition at line 34 of file classifier_tester.cpp.

{ CN_PRUNER, CN_FULL, CN_COUNT };

Function Documentation

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 101 of file classifier_tester.cpp.

                                {
  tesseract::CheckSharedLibraryVersion();
  ParseArguments(&argc, &argv);
  std::string file_prefix;
  auto trainer = tesseract::LoadTrainingData(argv + 1, false, nullptr, file_prefix);
  tesseract::TessBaseAPI *api;
  // Decode the classifier string.
  tesseract::ShapeClassifier *shape_classifier =
      InitializeClassifier(FLAGS_classifier.c_str(), trainer->unicharset(), argc, argv, &api);
  if (shape_classifier == nullptr) {
    fprintf(stderr, "Classifier init failed!:%s\n", FLAGS_classifier.c_str());
    return EXIT_FAILURE;
  }
 
  // We want to test junk as well if it is available.
  // trainer->IncludeJunk();
  // We want to test with replicated samples too.
  trainer->ReplicateAndRandomizeSamplesIfRequired();
 
  trainer->TestClassifierOnSamples(tesseract::CT_UNICHAR_TOP1_ERR,
                                   std::max(3, static_cast<int>(FLAGS_debug_level)), false,
                                   shape_classifier, nullptr);
  delete shape_classifier;
  delete api;
 
  return EXIT_SUCCESS;
} /* main */