#include <osdetect.h>

Public Member Functions
	OrientationDetector (const GenericVector< int > allowed_scripts, OSResults results)

bool	detect_blob (BLOB_CHOICE_LIST *scores)

int	get_orientation ()

Detailed Description

Definition at line 84 of file osdetect.h.

Constructor & Destructor Documentation

◆ OrientationDetector()

OrientationDetector::OrientationDetector	(	const GenericVector< int > *	allowed_scripts,
		OSResults *	results
	)

Definition at line 374 of file osdetect.cpp.

                                                                {
   osr_ = osr;
   allowed_scripts_ = allowed_scripts;
 }

Member Function Documentation

◆ detect_blob()

bool OrientationDetector::detect_blob ( BLOB_CHOICE_LIST * scores )

Definition at line 382 of file osdetect.cpp.

                                                               {
   float blob_o_score[4] = {0.0f, 0.0f, 0.0f, 0.0f};
   float total_blob_o_score = 0.0f;
  
   for (int i = 0; i < 4; ++i) {
     BLOB_CHOICE_IT choice_it(scores + i);
     if (!choice_it.empty()) {
       BLOB_CHOICE* choice = nullptr;
       if (allowed_scripts_ != nullptr && !allowed_scripts_->empty()) {
         // Find the top choice in an allowed script.
         for (choice_it.mark_cycle_pt(); !choice_it.cycled_list() &&
              choice == nullptr; choice_it.forward()) {
           int choice_script = choice_it.data()->script_id();
           int s = 0;
           for (s = 0; s < allowed_scripts_->size(); ++s) {
             if ((*allowed_scripts_)[s] == choice_script) {
               choice = choice_it.data();
               break;
             }
           }
         }
       } else {
         choice = choice_it.data();
       }
       if (choice != nullptr) {
         // The certainty score ranges between [-20,0]. This is converted here to
         // [0,1], with 1 indicating best match.
         blob_o_score[i] = 1 + 0.05 * choice->certainty();
         total_blob_o_score += blob_o_score[i];
       }
     }
   }
   if (total_blob_o_score == 0.0) return false;
   // Fill in any blanks with the worst score of the others. This is better than
   // picking an arbitrary probability for it and way better than -inf.
   float worst_score = 0.0f;
   int num_good_scores = 0;
   for (float f : blob_o_score) {
     if (f > 0.0f) {
       ++num_good_scores;
       if (worst_score == 0.0f || f < worst_score)
         worst_score = f;
     }
   }
   if (num_good_scores == 1) {
     // Lower worst if there is only one.
     worst_score /= 2.0f;
   }
   for (float& f : blob_o_score) {
     if (f == 0.0f) {
       f = worst_score;
       total_blob_o_score += worst_score;
     }
   }
   // Normalize the orientation scores for the blob and use them to
   // update the aggregated orientation score.
   for (int i = 0; total_blob_o_score != 0 && i < 4; ++i) {
     osr_->orientations[i] += log(blob_o_score[i] / total_blob_o_score);
   }
  
   // TODO(ranjith) Add an early exit test, based on min_orientation_margin,
   // as used in pagesegmain.cpp.
   return false;
 }

◆ get_orientation()

int OrientationDetector::get_orientation ( )

Definition at line 447 of file osdetect.cpp.

                                          {
   osr_->update_best_orientation();
   return osr_->best_result.orientation_id;
 }

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