tesseract::MockClassifier Class Reference

Inheritance diagram for tesseract::MockClassifier:

Public Member Functions
	MockClassifier (ShapeTable *shape_table)
virtual	~MockClassifier ()
virtual int	ClassifySample (const TrainingSample &sample, Pix *page_pix, int debug, UNICHAR_ID keep_this, GenericVector< ShapeRating > *results)
virtual const ShapeTable *	GetShapeTable () const
Public Member Functions inherited from tesseract::ShapeClassifier
virtual	~ShapeClassifier ()=default
virtual int	UnicharClassifySample (const TrainingSample &sample, Pix *page_pix, int debug, UNICHAR_ID keep_this, GenericVector< UnicharRating > *results)
virtual int	BestShapeForUnichar (const TrainingSample &sample, Pix *page_pix, UNICHAR_ID unichar_id, ShapeRating *result)
virtual const UNICHARSET &	GetUnicharset () const
virtual void	DebugDisplay (const TrainingSample &sample, Pix *page_pix, UNICHAR_ID unichar_id)
virtual int	DisplayClassifyAs (const TrainingSample &sample, Pix *page_pix, UNICHAR_ID unichar_id, int index, PointerVector< ScrollView > *windows)
virtual void	UnicharPrintResults (const char *context, const GenericVector< UnicharRating > &results) const
virtual void	PrintResults (const char *context, const GenericVector< ShapeRating > &results) const

Additional Inherited Members
Protected Member Functions inherited from tesseract::ShapeClassifier
void	FilterDuplicateUnichars (GenericVector< ShapeRating > *results) const

Detailed Description

Definition at line 70 of file mastertrainer_test.cc.

Constructor & Destructor Documentation

MockClassifier()

tesseract::MockClassifier::MockClassifier ( ShapeTable *  shape_table )

inlineexplicit

Definition at line 72 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()

virtual tesseract::MockClassifier::~MockClassifier ( )

inlinevirtual

Definition at line 80 of file mastertrainer_test.cc.

{}

Member Function Documentation

ClassifySample()

virtual int tesseract::MockClassifier::ClassifySample ( const TrainingSample &  sample, Pix *  page_pix, int  debug, UNICHAR_ID  keep_this, GenericVector< ShapeRating > *  results  )

inlinevirtual

Reimplemented from tesseract::ShapeClassifier.

Definition at line 88 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()

virtual const ShapeTable* tesseract::MockClassifier::GetShapeTable ( ) const

inlinevirtual

Implements tesseract::ShapeClassifier.

Definition at line 130 of file mastertrainer_test.cc.

{ return shape_table_; }

---

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

• /tesseract/unittest/mastertrainer_test.cc