mirror of
https://github.com/tesseract-ocr/tesseract.git
synced 2024-12-11 15:09:03 +08:00
53fc4456cc
Eliminated the flexfx scheme for calling global feature extractor functions through an array of function pointers. Deleted dead code I found as a by-product. This CL does not change BlobToTrainingSample or ExtractFeatures to be full members of Classify (the eventual goal) as that would make it even bigger, since there are a lot of callers to these functions. When ExtractFeatures and BlobToTrainingSample are members of Classify they will be able to access control parameters in Classify, which will greatly simplify developing variations to the feature extraction process.
109 lines
4.5 KiB
C++
109 lines
4.5 KiB
C++
/******************************************************************************
|
|
** Filename: blobclass.c
|
|
** Purpose: High level blob classification and training routines.
|
|
** Author: Dan Johnson
|
|
** History: 7/21/89, DSJ, Created.
|
|
**
|
|
** (c) Copyright Hewlett-Packard Company, 1988.
|
|
** Licensed under the Apache License, Version 2.0 (the "License");
|
|
** you may not use this file except in compliance with the License.
|
|
** You may obtain a copy of the License at
|
|
** http://www.apache.org/licenses/LICENSE-2.0
|
|
** Unless required by applicable law or agreed to in writing, software
|
|
** distributed under the License is distributed on an "AS IS" BASIS,
|
|
** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
** See the License for the specific language governing permissions and
|
|
** limitations under the License.
|
|
******************************************************************************/
|
|
|
|
/**----------------------------------------------------------------------------
|
|
Include Files and Type Defines
|
|
----------------------------------------------------------------------------**/
|
|
#include "blobclass.h"
|
|
|
|
#include <stdio.h>
|
|
|
|
#include "classify.h"
|
|
#include "efio.h"
|
|
#include "featdefs.h"
|
|
#include "mf.h"
|
|
#include "normfeat.h"
|
|
|
|
static const char kUnknownFontName[] = "UnknownFont";
|
|
|
|
STRING_VAR(classify_font_name, kUnknownFontName,
|
|
"Default font name to be used in training");
|
|
|
|
namespace tesseract {
|
|
/**----------------------------------------------------------------------------
|
|
Public Code
|
|
----------------------------------------------------------------------------**/
|
|
// Finds the name of the training font and returns it in fontname, by cutting
|
|
// it out based on the expectation that the filename is of the form:
|
|
// /path/to/dir/[lang].[fontname].exp[num]
|
|
// The [lang], [fontname] and [num] fields should not have '.' characters.
|
|
// If the global parameter classify_font_name is set, its value is used instead.
|
|
void ExtractFontName(const STRING& filename, STRING* fontname) {
|
|
*fontname = classify_font_name;
|
|
if (*fontname == kUnknownFontName) {
|
|
// filename is expected to be of the form [lang].[fontname].exp[num]
|
|
// The [lang], [fontname] and [num] fields should not have '.' characters.
|
|
const char *basename = strrchr(filename.string(), '/');
|
|
const char *firstdot = strchr(basename ? basename : filename.string(), '.');
|
|
const char *lastdot = strrchr(filename.string(), '.');
|
|
if (firstdot != lastdot && firstdot != NULL && lastdot != NULL) {
|
|
++firstdot;
|
|
*fontname = firstdot;
|
|
fontname->truncate_at(lastdot - firstdot);
|
|
}
|
|
}
|
|
}
|
|
|
|
/*---------------------------------------------------------------------------*/
|
|
// Extracts features from the given blob and saves them in the tr_file_data_
|
|
// member variable.
|
|
// fontname: Name of font that this blob was printed in.
|
|
// cn_denorm: Character normalization transformation to apply to the blob.
|
|
// fx_info: Character normalization parameters computed with cn_denorm.
|
|
// blob_text: Ground truth text for the blob.
|
|
void Classify::LearnBlob(const STRING& fontname, TBLOB* blob,
|
|
const DENORM& cn_denorm,
|
|
const INT_FX_RESULT_STRUCT& fx_info,
|
|
const char* blob_text) {
|
|
CHAR_DESC CharDesc = NewCharDescription(feature_defs_);
|
|
CharDesc->FeatureSets[0] = ExtractMicros(blob, cn_denorm);
|
|
CharDesc->FeatureSets[1] = ExtractCharNormFeatures(fx_info);
|
|
CharDesc->FeatureSets[2] = ExtractIntCNFeatures(*blob, fx_info);
|
|
CharDesc->FeatureSets[3] = ExtractIntGeoFeatures(*blob, fx_info);
|
|
|
|
if (ValidCharDescription(feature_defs_, CharDesc)) {
|
|
// Label the features with a class name and font name.
|
|
tr_file_data_ += "\n";
|
|
tr_file_data_ += fontname;
|
|
tr_file_data_ += " ";
|
|
tr_file_data_ += blob_text;
|
|
tr_file_data_ += "\n";
|
|
|
|
// write micro-features to file and clean up
|
|
WriteCharDescription(feature_defs_, CharDesc, &tr_file_data_);
|
|
} else {
|
|
tprintf("Blob learned was invalid!\n");
|
|
}
|
|
FreeCharDescription(CharDesc);
|
|
} // LearnBlob
|
|
|
|
// Writes stored training data to a .tr file based on the given filename.
|
|
// Returns false on error.
|
|
bool Classify::WriteTRFile(const STRING& filename) {
|
|
STRING tr_filename = filename + ".tr";
|
|
FILE* fp = Efopen(tr_filename.string(), "wb");
|
|
int len = tr_file_data_.length();
|
|
bool result =
|
|
fwrite(&tr_file_data_[0], sizeof(tr_file_data_[0]), len, fp) == len;
|
|
fclose(fp);
|
|
tr_file_data_.truncate_at(0);
|
|
return result;
|
|
}
|
|
|
|
} // namespace tesseract.
|