tesseract  3.04.00
tesseract::ConvNetCharClassifier Class Reference

#include <conv_net_classifier.h>

Inheritance diagram for tesseract::ConvNetCharClassifier:
tesseract::CharClassifier

Public Member Functions

 ConvNetCharClassifier (CharSet *char_set, TuningParams *params, FeatureBase *feat_extract)
 
virtual ~ConvNetCharClassifier ()
 
virtual bool Train (CharSamp *char_samp, int ClassID)
 
virtual bool SetLearnParam (char *var_name, float val)
 
void SetNet (tesseract::NeuralNet *net)
 
virtual CharAltListClassify (CharSamp *char_samp)
 
virtual int CharCost (CharSamp *char_samp)
 
- Public Member Functions inherited from tesseract::CharClassifier
 CharClassifier (CharSet *char_set, TuningParams *params, FeatureBase *feat_extract)
 
virtual ~CharClassifier ()
 
FeatureBaseFeatureExtractor ()
 
bool CaseSensitive () const
 
void SetCaseSensitive (bool case_sensitive)
 

Additional Inherited Members

- Protected Attributes inherited from tesseract::CharClassifier
FeatureBasefeat_extract_
 
CharSetchar_set_
 
TuningParamsparams_
 
int ** fold_sets_
 
int * fold_set_len_
 
int fold_set_cnt_
 
bool init_
 
bool case_sensitive_
 

Detailed Description

Definition at line 48 of file conv_net_classifier.h.

Constructor & Destructor Documentation

tesseract::ConvNetCharClassifier::ConvNetCharClassifier ( CharSet char_set,
TuningParams params,
FeatureBase feat_extract 
)

Definition at line 38 of file conv_net_classifier.cpp.

41  : CharClassifier(char_set, params, feat_extract) {
42  char_net_ = NULL;
43  net_input_ = NULL;
44  net_output_ = NULL;
45 }
CharClassifier(CharSet *char_set, TuningParams *params, FeatureBase *feat_extract)
#define NULL
Definition: host.h:144
tesseract::ConvNetCharClassifier::~ConvNetCharClassifier ( )
virtual

Definition at line 47 of file conv_net_classifier.cpp.

47  {
48  if (char_net_ != NULL) {
49  delete char_net_;
50  char_net_ = NULL;
51  }
52 
53  if (net_input_ != NULL) {
54  delete []net_input_;
55  net_input_ = NULL;
56  }
57 
58  if (net_output_ != NULL) {
59  delete []net_output_;
60  net_output_ = NULL;
61  }
62 }
#define NULL
Definition: host.h:144

Member Function Documentation

int tesseract::ConvNetCharClassifier::CharCost ( CharSamp char_samp)
virtual

Implements tesseract::CharClassifier.

Definition at line 177 of file conv_net_classifier.cpp.

177  {
178  if (RunNets(char_samp) == false) {
179  return 0;
180  }
181  return CubeUtils::Prob2Cost(1.0f - net_output_[0]);
182 }
static int Prob2Cost(double prob_val)
Definition: cube_utils.cpp:35
CharAltList * tesseract::ConvNetCharClassifier::Classify ( CharSamp char_samp)
virtual

Implements tesseract::CharClassifier.

Definition at line 186 of file conv_net_classifier.cpp.

186  {
187  // run the needed nets
188  if (RunNets(char_samp) == false) {
189  return NULL;
190  }
191 
192  int class_cnt = char_set_->ClassCount();
193 
194  // create an altlist
195  CharAltList *alt_list = new CharAltList(char_set_, class_cnt);
196  if (alt_list == NULL) {
197  fprintf(stderr, "Cube WARNING (ConvNetCharClassifier::Classify): "
198  "returning emtpy CharAltList\n");
199  return NULL;
200  }
201 
202  for (int out = 1; out < class_cnt; out++) {
203  int cost = CubeUtils::Prob2Cost(net_output_[out]);
204  alt_list->Insert(out, cost);
205  }
206 
207  return alt_list;
208 }
#define NULL
Definition: host.h:144
static int Prob2Cost(double prob_val)
Definition: cube_utils.cpp:35
int ClassCount() const
Definition: char_set.h:111
bool tesseract::ConvNetCharClassifier::SetLearnParam ( char *  var_name,
float  val 
)
virtual

Implements tesseract::CharClassifier.

Definition at line 74 of file conv_net_classifier.cpp.

74  {
75  // TODO(ahmadab): implementation of parameter initializing.
76  return false;
77 }
void tesseract::ConvNetCharClassifier::SetNet ( tesseract::NeuralNet net)

Definition at line 211 of file conv_net_classifier.cpp.

211  {
212  if (char_net_ != NULL) {
213  delete char_net_;
214  char_net_ = NULL;
215  }
216  char_net_ = char_net;
217 }
#define NULL
Definition: host.h:144
bool tesseract::ConvNetCharClassifier::Train ( CharSamp char_samp,
int  ClassID 
)
virtual

Implements tesseract::CharClassifier.

Definition at line 67 of file conv_net_classifier.cpp.

67  {
68  return false;
69 }

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