Inheritance diagram for MockClassifier:

Public Member Functions
	MockClassifier (ShapeTable *shape_table)

	~MockClassifier () override=default

int	ClassifySample (const TrainingSample &sample, Image page_pix, int debug, UNICHAR_ID keep_this, std::vector< ShapeRating > *results) override

const ShapeTable *	GetShapeTable () const override

Public Member Functions inherited from tesseract::ShapeClassifier
virtual	~ShapeClassifier ()=default

virtual int	UnicharClassifySample (const TrainingSample &sample, Image page_pix, int debug, UNICHAR_ID keep_this, std::vector< UnicharRating > *results)

virtual int	BestShapeForUnichar (const TrainingSample &sample, Image page_pix, UNICHAR_ID unichar_id, ShapeRating *result)

virtual const ShapeTable *	GetShapeTable () const =0

virtual const UNICHARSET &	GetUnicharset () const

void	DebugDisplay (const TrainingSample &sample, Image page_pix, UNICHAR_ID unichar_id)

virtual int	DisplayClassifyAs (const TrainingSample &sample, Image page_pix, UNICHAR_ID unichar_id, int index, std::vector< ScrollView * > &windows)

virtual void	UnicharPrintResults (const char *context, const std::vector< UnicharRating > &results) const

virtual void	PrintResults (const char *context, const std::vector< ShapeRating > &results) const

Additional Inherited Members
Protected Member Functions inherited from tesseract::ShapeClassifier
virtual int	ClassifySample (const TrainingSample &sample, Image page_pix, int debug, UNICHAR_ID keep_this, std::vector< ShapeRating > *results)

void	FilterDuplicateUnichars (std::vector< ShapeRating > *results) const

Detailed Description

Definition at line 63 of file mastertrainer_test.cc.

Constructor & Destructor Documentation

◆ MockClassifier()

MockClassifier::MockClassifier ( ShapeTable * shape_table )

inlineexplicit

Definition at line 65 of file mastertrainer_test.cc.

      : shape_table_(shape_table), num_done_(0), done_bad_font_(false) {
    // Add a false font answer to the shape table. We pick a random unichar_id,
    // add a new shape for it with a false font. Font must actually exist in
    // the font table, but not match anything in the first 1000 samples.
    false_unichar_id_ = 67;
    false_shape_ = shape_table_->AddShape(false_unichar_id_, 25);
  }

◆ ~MockClassifier()

MockClassifier::~MockClassifier ( )

overridedefault

Member Function Documentation

◆ ClassifySample()

int MockClassifier::ClassifySample	(	const TrainingSample &	sample,
		Image	page_pix,
		int	debug,
		UNICHAR_ID	keep_this,
		std::vector< ShapeRating > *	results
	)

inlineoverridevirtual

Reimplemented from tesseract::ShapeClassifier.

Definition at line 81 of file mastertrainer_test.cc.

                                                               {
    results->clear();
    // Everything except the first kNumNonReject is a reject.
    if (++num_done_ > kNumNonReject) {
      return 0;
    }
 
    int class_id = sample.class_id();
    int font_id = sample.font_id();
    int shape_id = shape_table_->FindShape(class_id, font_id);
    // Get ids of some wrong answers.
    int wrong_id1 = shape_id > 10 ? shape_id - 1 : shape_id + 1;
    int wrong_id2 = shape_id > 10 ? shape_id - 2 : shape_id + 2;
    if (num_done_ <= kNumTopNErrs) {
      // The first kNumTopNErrs are top-n errors.
      results->push_back(ShapeRating(wrong_id1, 1.0f));
    } else if (num_done_ <= kNumTop2Errs) {
      // The next kNumTop2Errs - kNumTopNErrs are top-2 errors.
      results->push_back(ShapeRating(wrong_id1, 1.0f));
      results->push_back(ShapeRating(wrong_id2, 0.875f));
      results->push_back(ShapeRating(shape_id, 0.75f));
    } else if (num_done_ <= kNumTop1Errs) {
      // The next kNumTop1Errs - kNumTop2Errs are top-1 errors.
      results->push_back(ShapeRating(wrong_id1, 1.0f));
      results->push_back(ShapeRating(shape_id, 0.8f));
    } else if (num_done_ <= kNumTopTopErrs) {
      // The next kNumTopTopErrs - kNumTop1Errs are cases where the actual top
      // is not correct, but do not count as a top-1 error because the rating
      // is close enough to the top answer.
      results->push_back(ShapeRating(wrong_id1, 1.0f));
      results->push_back(ShapeRating(shape_id, 0.99f));
    } else if (!done_bad_font_ && class_id == false_unichar_id_) {
      // There is a single character with a bad font.
      results->push_back(ShapeRating(false_shape_, 1.0f));
      done_bad_font_ = true;
    } else {
      // Everything else is correct.
      results->push_back(ShapeRating(shape_id, 1.0f));
    }
    return results->size();
  }

◆ GetShapeTable()

const ShapeTable * MockClassifier::GetShapeTable ( ) const

inlineoverridevirtual

Implements tesseract::ShapeClassifier.

Definition at line 124 of file mastertrainer_test.cc.

                                                   {
    return shape_table_;
  }

The documentation for this class was generated from the following file:

/media/home/debian/src/github/tesseract-ocr/tesseract/unittest/mastertrainer_test.cc

Public Member Functions