tesseract::ConvNetCharClassifier Class Reference

#include <conv_net_classifier.h>

Inheritance diagram for tesseract::ConvNetCharClassifier:

List of all members.

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 CharAltList *	Classify (CharSamp *char_samp)
virtual int	CharCost (CharSamp *char_samp)

---

Detailed Description

Definition at line 48 of file conv_net_classifier.h.

---

Constructor & Destructor Documentation

usr src packages BUILD tesseract cube conv_net_classifier cpp tesseract::ConvNetCharClassifier::ConvNetCharClassifier	(	CharSet *	char_set,
		TuningParams *	params,
		FeatureBase *	feat_extract
	)

Definition at line 39 of file conv_net_classifier.cpp.

    : CharClassifier(char_set, params, feat_extract) {
  char_net_ = NULL;
  net_input_ = NULL;
  net_output_ = NULL;
}

tesseract::ConvNetCharClassifier::~ConvNetCharClassifier

(

)

[virtual]

Definition at line 48 of file conv_net_classifier.cpp.

                         {
    delete char_net_;
    char_net_ = NULL;
  }

  if (net_input_ != NULL) {
    delete []net_input_;
    net_input_ = NULL;
  }

  if (net_output_ != NULL) {
    delete []net_output_;
    net_output_ = NULL;
  }
}

---

Member Function Documentation

int tesseract::ConvNetCharClassifier::CharCost

(

CharSamp *

char_samp

)

[virtual]

return the cost of being a char

Implements tesseract::CharClassifier.

Definition at line 188 of file conv_net_classifier.cpp.

                                   {
    return 0;
  }
  return CubeUtils::Prob2Cost(1.0f - net_output_[0]);
}

CharAltList * tesseract::ConvNetCharClassifier::Classify

(

CharSamp *

char_samp

)

[virtual]

classifies a charsamp and returns an alternate list of chars sorted by char costs

Implements tesseract::CharClassifier.

Definition at line 199 of file conv_net_classifier.cpp.

                                   {
    return NULL;
  }

  int class_cnt = char_set_->ClassCount();

  // create an altlist
  CharAltList *alt_list = new CharAltList(char_set_, class_cnt);
  if (alt_list == NULL) {
    fprintf(stderr, "Cube WARNING (ConvNetCharClassifier::Classify): "
            "returning emtpy CharAltList\n");
    return NULL;
  }

  for (int out = 1; out < class_cnt; out++) {
    int cost = CubeUtils::Prob2Cost(net_output_[out]);
    alt_list->Insert(out, cost);
  }

  return alt_list;
}

bool tesseract::ConvNetCharClassifier::SetLearnParam	(	char *	var_name,
		float	val
	)		[virtual]

A secondary function needed for training. Allows the trainer to set the value of any train-time parameter. This function is currently not implemented. TODO(ahmadab): implement end-2-end training

Implements tesseract::CharClassifier.

Definition at line 79 of file conv_net_classifier.cpp.

void tesseract::ConvNetCharClassifier::SetNet

(

tesseract::NeuralNet *

char_net

)

Set an external net (for training purposes)

Definition at line 226 of file conv_net_classifier.cpp.

                         {
    delete char_net_;
    char_net_ = NULL;
  }
  char_net_ = char_net;
}

bool tesseract::ConvNetCharClassifier::Train	(	CharSamp *	char_samp,
		int	ClassID
	)		[virtual]

The main training function. Given a sample and a class ID the classifier updates its parameters according to its learning algorithm. This function is currently not implemented. TODO(ahmadab): implement end-2-end training

Implements tesseract::CharClassifier.

Definition at line 70 of file conv_net_classifier.cpp.

---

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

• cube/conv_net_classifier.h
• cube/conv_net_classifier.cpp