|
Tesseract
3.02
|
#include <conv_net_classifier.h>
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) |
Definition at line 48 of file conv_net_classifier.h.
| tesseract::ConvNetCharClassifier::ConvNetCharClassifier | ( | CharSet * | char_set, |
| TuningParams * | params, | ||
| FeatureBase * | feat_extract | ||
| ) |
Definition at line 38 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] |
| int tesseract::ConvNetCharClassifier::CharCost | ( | CharSamp * | char_samp | ) | [virtual] |
Implements tesseract::CharClassifier.
Definition at line 177 of file conv_net_classifier.cpp.
{
if (RunNets(char_samp) == false) {
return 0;
}
return CubeUtils::Prob2Cost(1.0f - net_output_[0]);
}
| CharAltList * tesseract::ConvNetCharClassifier::Classify | ( | CharSamp * | char_samp | ) | [virtual] |
Implements tesseract::CharClassifier.
Definition at line 186 of file conv_net_classifier.cpp.
{
// run the needed nets
if (RunNets(char_samp) == false) {
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] |
Implements tesseract::CharClassifier.
Definition at line 74 of file conv_net_classifier.cpp.
{
// TODO(ahmadab): implementation of parameter initializing.
return false;
}
| void tesseract::ConvNetCharClassifier::SetNet | ( | tesseract::NeuralNet * | net | ) |
Definition at line 211 of file conv_net_classifier.cpp.
| bool tesseract::ConvNetCharClassifier::Train | ( | CharSamp * | char_samp, |
| int | ClassID | ||
| ) | [virtual] |
Implements tesseract::CharClassifier.
Definition at line 67 of file conv_net_classifier.cpp.
{
return false;
}