mirror of
https://github.com/tesseract-ocr/tesseract.git
synced 2024-12-18 11:28:51 +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.
290 lines
10 KiB
C++
290 lines
10 KiB
C++
/******************************************************************************
|
|
** Filename: picofeat.c
|
|
** Purpose: Definition of pico-features.
|
|
** Author: Dan Johnson
|
|
** History: 9/4/90, 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 "picofeat.h"
|
|
|
|
#include "classify.h"
|
|
#include "efio.h"
|
|
#include "featdefs.h"
|
|
#include "fpoint.h"
|
|
#include "mfoutline.h"
|
|
#include "ocrfeatures.h"
|
|
#include "params.h"
|
|
#include "trainingsample.h"
|
|
|
|
#include <math.h>
|
|
#include <stdio.h>
|
|
|
|
/*---------------------------------------------------------------------------
|
|
Variables
|
|
----------------------------------------------------------------------------*/
|
|
|
|
double_VAR(classify_pico_feature_length, 0.05, "Pico Feature Length");
|
|
|
|
/*---------------------------------------------------------------------------
|
|
Private Function Prototypes
|
|
----------------------------------------------------------------------------*/
|
|
void ConvertSegmentToPicoFeat(FPOINT *Start,
|
|
FPOINT *End,
|
|
FEATURE_SET FeatureSet);
|
|
|
|
void ConvertToPicoFeatures2(MFOUTLINE Outline, FEATURE_SET FeatureSet);
|
|
|
|
void NormalizePicoX(FEATURE_SET FeatureSet);
|
|
|
|
/**----------------------------------------------------------------------------
|
|
Public Code
|
|
----------------------------------------------------------------------------**/
|
|
/*---------------------------------------------------------------------------*/
|
|
namespace tesseract {
|
|
FEATURE_SET Classify::ExtractPicoFeatures(TBLOB *Blob) {
|
|
/*
|
|
** Parameters:
|
|
** Blob blob to extract pico-features from
|
|
** LineStats statistics on text row blob is in
|
|
** Globals:
|
|
** classify_norm_method normalization method currently specified
|
|
** Operation: Dummy for now.
|
|
** Return: Pico-features for Blob.
|
|
** Exceptions: none
|
|
** History: 9/4/90, DSJ, Created.
|
|
*/
|
|
LIST Outlines;
|
|
LIST RemainingOutlines;
|
|
MFOUTLINE Outline;
|
|
FEATURE_SET FeatureSet;
|
|
FLOAT32 XScale, YScale;
|
|
|
|
FeatureSet = NewFeatureSet(MAX_PICO_FEATURES);
|
|
Outlines = ConvertBlob(Blob);
|
|
NormalizeOutlines(Outlines, &XScale, &YScale);
|
|
RemainingOutlines = Outlines;
|
|
iterate(RemainingOutlines) {
|
|
Outline = (MFOUTLINE) first_node (RemainingOutlines);
|
|
ConvertToPicoFeatures2(Outline, FeatureSet);
|
|
}
|
|
if (classify_norm_method == baseline)
|
|
NormalizePicoX(FeatureSet);
|
|
FreeOutlines(Outlines);
|
|
return (FeatureSet);
|
|
|
|
} /* ExtractPicoFeatures */
|
|
} // namespace tesseract
|
|
|
|
/**----------------------------------------------------------------------------
|
|
Private Code
|
|
----------------------------------------------------------------------------**/
|
|
/*---------------------------------------------------------------------------*/
|
|
void ConvertSegmentToPicoFeat(FPOINT *Start,
|
|
FPOINT *End,
|
|
FEATURE_SET FeatureSet) {
|
|
/*
|
|
** Parameters:
|
|
** Start starting point of pico-feature
|
|
** End ending point of pico-feature
|
|
** FeatureSet set to add pico-feature to
|
|
** Globals:
|
|
** classify_pico_feature_length length of a single pico-feature
|
|
** Operation: This routine converts an entire segment of an outline
|
|
** into a set of pico features which are added to
|
|
** FeatureSet. The length of the segment is rounded to the
|
|
** nearest whole number of pico-features. The pico-features
|
|
** are spaced evenly over the entire segment.
|
|
** Return: none (results are placed in FeatureSet)
|
|
** Exceptions: none
|
|
** History: Tue Apr 30 15:44:34 1991, DSJ, Created.
|
|
*/
|
|
FEATURE Feature;
|
|
FLOAT32 Angle;
|
|
FLOAT32 Length;
|
|
int NumFeatures;
|
|
FPOINT Center;
|
|
FPOINT Delta;
|
|
int i;
|
|
|
|
Angle = NormalizedAngleFrom (Start, End, 1.0);
|
|
Length = DistanceBetween (*Start, *End);
|
|
NumFeatures = (int) floor (Length / classify_pico_feature_length + 0.5);
|
|
if (NumFeatures < 1)
|
|
NumFeatures = 1;
|
|
|
|
/* compute vector for one pico feature */
|
|
Delta.x = XDelta (*Start, *End) / NumFeatures;
|
|
Delta.y = YDelta (*Start, *End) / NumFeatures;
|
|
|
|
/* compute position of first pico feature */
|
|
Center.x = Start->x + Delta.x / 2.0;
|
|
Center.y = Start->y + Delta.y / 2.0;
|
|
|
|
/* compute each pico feature in segment and add to feature set */
|
|
for (i = 0; i < NumFeatures; i++) {
|
|
Feature = NewFeature (&PicoFeatDesc);
|
|
Feature->Params[PicoFeatDir] = Angle;
|
|
Feature->Params[PicoFeatX] = Center.x;
|
|
Feature->Params[PicoFeatY] = Center.y;
|
|
AddFeature(FeatureSet, Feature);
|
|
|
|
Center.x += Delta.x;
|
|
Center.y += Delta.y;
|
|
}
|
|
} /* ConvertSegmentToPicoFeat */
|
|
|
|
|
|
/*---------------------------------------------------------------------------*/
|
|
void ConvertToPicoFeatures2(MFOUTLINE Outline, FEATURE_SET FeatureSet) {
|
|
/*
|
|
** Parameters:
|
|
** Outline outline to extract micro-features from
|
|
** FeatureSet set of features to add pico-features to
|
|
** Globals:
|
|
** classify_pico_feature_length
|
|
** length of features to be extracted
|
|
** Operation:
|
|
** This routine steps thru the specified outline and cuts it
|
|
** up into pieces of equal length. These pieces become the
|
|
** desired pico-features. Each segment in the outline
|
|
** is converted into an integral number of pico-features.
|
|
** Return: none (results are returned in FeatureSet)
|
|
** Exceptions: none
|
|
** History: 4/30/91, DSJ, Adapted from ConvertToPicoFeatures().
|
|
*/
|
|
MFOUTLINE Next;
|
|
MFOUTLINE First;
|
|
MFOUTLINE Current;
|
|
|
|
if (DegenerateOutline(Outline))
|
|
return;
|
|
|
|
First = Outline;
|
|
Current = First;
|
|
Next = NextPointAfter(Current);
|
|
do {
|
|
/* note that an edge is hidden if the ending point of the edge is
|
|
marked as hidden. This situation happens because the order of
|
|
the outlines is reversed when they are converted from the old
|
|
format. In the old format, a hidden edge is marked by the
|
|
starting point for that edge. */
|
|
if (!(PointAt(Next)->Hidden))
|
|
ConvertSegmentToPicoFeat (&(PointAt(Current)->Point),
|
|
&(PointAt(Next)->Point), FeatureSet);
|
|
|
|
Current = Next;
|
|
Next = NextPointAfter(Current);
|
|
}
|
|
while (Current != First);
|
|
|
|
} /* ConvertToPicoFeatures2 */
|
|
|
|
|
|
/*---------------------------------------------------------------------------*/
|
|
void NormalizePicoX(FEATURE_SET FeatureSet) {
|
|
/*
|
|
** Parameters:
|
|
** FeatureSet pico-features to be normalized
|
|
** Globals: none
|
|
** Operation: This routine computes the average x position over all
|
|
** of the pico-features in FeatureSet and then renormalizes
|
|
** the pico-features to force this average to be the x origin
|
|
** (i.e. x=0).
|
|
** Return: none (FeatureSet is changed)
|
|
** Exceptions: none
|
|
** History: Tue Sep 4 16:50:08 1990, DSJ, Created.
|
|
*/
|
|
int i;
|
|
FEATURE Feature;
|
|
FLOAT32 Origin = 0.0;
|
|
|
|
for (i = 0; i < FeatureSet->NumFeatures; i++) {
|
|
Feature = FeatureSet->Features[i];
|
|
Origin += Feature->Params[PicoFeatX];
|
|
}
|
|
Origin /= FeatureSet->NumFeatures;
|
|
|
|
for (i = 0; i < FeatureSet->NumFeatures; i++) {
|
|
Feature = FeatureSet->Features[i];
|
|
Feature->Params[PicoFeatX] -= Origin;
|
|
}
|
|
} /* NormalizePicoX */
|
|
|
|
namespace tesseract {
|
|
/*---------------------------------------------------------------------------*/
|
|
FEATURE_SET Classify::ExtractIntCNFeatures(
|
|
const TBLOB& blob, const INT_FX_RESULT_STRUCT& fx_info) {
|
|
/*
|
|
** Parameters:
|
|
** blob blob to extract features from
|
|
** denorm normalization/denormalization parameters.
|
|
** Return: Integer character-normalized features for blob.
|
|
** Exceptions: none
|
|
** History: 8/8/2011, rays, Created.
|
|
*/
|
|
INT_FX_RESULT_STRUCT local_fx_info(fx_info);
|
|
GenericVector<INT_FEATURE_STRUCT> bl_features;
|
|
tesseract::TrainingSample* sample = tesseract::BlobToTrainingSample(
|
|
blob, false, &local_fx_info, &bl_features);
|
|
if (sample == NULL) return NULL;
|
|
|
|
int num_features = sample->num_features();
|
|
const INT_FEATURE_STRUCT* features = sample->features();
|
|
FEATURE_SET feature_set = NewFeatureSet(num_features);
|
|
for (int f = 0; f < num_features; ++f) {
|
|
FEATURE feature = NewFeature(&IntFeatDesc);
|
|
|
|
feature->Params[IntX] = features[f].X;
|
|
feature->Params[IntY] = features[f].Y;
|
|
feature->Params[IntDir] = features[f].Theta;
|
|
AddFeature(feature_set, feature);
|
|
}
|
|
delete sample;
|
|
|
|
return feature_set;
|
|
} /* ExtractIntCNFeatures */
|
|
|
|
/*---------------------------------------------------------------------------*/
|
|
FEATURE_SET Classify::ExtractIntGeoFeatures(
|
|
const TBLOB& blob, const INT_FX_RESULT_STRUCT& fx_info) {
|
|
/*
|
|
** Parameters:
|
|
** blob blob to extract features from
|
|
** denorm normalization/denormalization parameters.
|
|
** Return: Geometric (top/bottom/width) features for blob.
|
|
** Exceptions: none
|
|
** History: 8/8/2011, rays, Created.
|
|
*/
|
|
INT_FX_RESULT_STRUCT local_fx_info(fx_info);
|
|
GenericVector<INT_FEATURE_STRUCT> bl_features;
|
|
tesseract::TrainingSample* sample = tesseract::BlobToTrainingSample(
|
|
blob, false, &local_fx_info, &bl_features);
|
|
if (sample == NULL) return NULL;
|
|
|
|
FEATURE_SET feature_set = NewFeatureSet(1);
|
|
FEATURE feature = NewFeature(&IntFeatDesc);
|
|
|
|
feature->Params[GeoBottom] = sample->geo_feature(GeoBottom);
|
|
feature->Params[GeoTop] = sample->geo_feature(GeoTop);
|
|
feature->Params[GeoWidth] = sample->geo_feature(GeoWidth);
|
|
AddFeature(feature_set, feature);
|
|
delete sample;
|
|
|
|
return feature_set;
|
|
} /* ExtractIntGeoFeatures */
|
|
|
|
} // namespace tesseract.
|