tesseract::Convolve Class Reference

#include <convolve.h>

Inheritance diagram for tesseract::Convolve:

Public Member Functions
	Convolve (const STRING &name, int ni, int half_x, int half_y)
virtual	~Convolve ()
virtual STRING	spec () const
virtual bool	Serialize (TFile *fp) const
virtual bool	DeSerialize (TFile *fp)
virtual void	Forward (bool debug, const NetworkIO &input, const TransposedArray *input_transpose, NetworkScratch *scratch, NetworkIO *output)
virtual bool	Backward (bool debug, const NetworkIO &fwd_deltas, NetworkScratch *scratch, NetworkIO *back_deltas)
Public Member Functions inherited from tesseract::Network
	Network ()
	Network (NetworkType type, const STRING &name, int ni, int no)
virtual	~Network ()
NetworkType	type () const
bool	IsTraining () const
bool	needs_to_backprop () const
int	num_weights () const
int	NumInputs () const
int	NumOutputs () const
virtual StaticShape	InputShape () const
virtual StaticShape	OutputShape (const StaticShape &input_shape) const
const STRING &	name () const
bool	TestFlag (NetworkFlags flag) const
virtual bool	IsPlumbingType () const
virtual void	SetEnableTraining (TrainingState state)
virtual void	SetNetworkFlags (uinT32 flags)
virtual int	InitWeights (float range, TRand *randomizer)
virtual void	ConvertToInt ()
virtual void	SetRandomizer (TRand *randomizer)
virtual bool	SetupNeedsBackprop (bool needs_backprop)
virtual int	XScaleFactor () const
virtual void	CacheXScaleFactor (int factor)
virtual void	DebugWeights ()
virtual void	Update (float learning_rate, float momentum, int num_samples)
virtual void	CountAlternators (const Network &other, double *same, double *changed) const
void	DisplayForward (const NetworkIO &matrix)
void	DisplayBackward (const NetworkIO &matrix)

Protected Attributes
inT32	half_x_
inT32	half_y_
Protected Attributes inherited from tesseract::Network
NetworkType	type_
TrainingState	training_
bool	needs_to_backprop_
inT32	network_flags_
inT32	ni_
inT32	no_
inT32	num_weights_
STRING	name_
ScrollView *	forward_win_
ScrollView *	backward_win_
TRand *	randomizer_

Additional Inherited Members
Static Public Member Functions inherited from tesseract::Network
static Network *	CreateFromFile (TFile *fp)
static void	ClearWindow (bool tess_coords, const char *window_name, int width, int height, ScrollView **window)
static int	DisplayImage (Pix *pix, ScrollView *window)
Protected Member Functions inherited from tesseract::Network
double	Random (double range)
Static Protected Attributes inherited from tesseract::Network
static char const *const	kTypeNames [NT_COUNT]

Detailed Description

Definition at line 33 of file convolve.h.

Constructor & Destructor Documentation

Convolve()

tesseract::Convolve::Convolve	(	const STRING &	name,
		int	ni,
		int	half_x,
		int	half_y
	)

Definition at line 28 of file convolve.cpp.

  : Network(NT_CONVOLVE, name, ni, ni * (2*half_x + 1) * (2*half_y + 1)),
    half_x_(half_x), half_y_(half_y) {
}

~Convolve()

tesseract::Convolve::~Convolve ( )

virtual

Definition at line 33 of file convolve.cpp.

                    {
}

Member Function Documentation

Backward()

bool tesseract::Convolve::Backward ( bool  debug, const NetworkIO &  fwd_deltas, NetworkScratch *  scratch, NetworkIO *  back_deltas  )

virtual

Reimplemented from tesseract::Network.

Definition at line 88 of file convolve.cpp.

                                                {
  back_deltas->Resize(fwd_deltas, ni_);
  NetworkScratch::IO delta_sum;
  delta_sum.ResizeFloat(fwd_deltas, ni_, scratch);
  delta_sum->Zero();
  int y_scale = 2 * half_y_ + 1;
  StrideMap::Index src_index(fwd_deltas.stride_map());
  do {
    // Stack x_scale groups of y_scale * ni_ inputs together.
    int t = src_index.t();
    int out_ix = 0;
    for (int x = -half_x_; x <= half_x_; ++x, out_ix += y_scale * ni_) {
      StrideMap::Index x_index(src_index);
      if (x_index.AddOffset(x, FD_WIDTH)) {
        int out_iy = out_ix;
        for (int y = -half_y_; y <= half_y_; ++y, out_iy += ni_) {
          StrideMap::Index y_index(x_index);
          if (y_index.AddOffset(y, FD_HEIGHT)) {
            fwd_deltas.AddTimeStepPart(t, out_iy, ni_,
                                       delta_sum->f(y_index.t()));
          }
        }
      }
    }
  } while (src_index.Increment());
  back_deltas->CopyWithNormalization(*delta_sum, fwd_deltas);
  return true;
}

DeSerialize()

bool tesseract::Convolve::DeSerialize ( TFile *  fp )

virtual

Reimplemented from tesseract::Network.

Definition at line 45 of file convolve.cpp.

                                    {
  if (fp->FReadEndian(&half_x_, sizeof(half_x_), 1) != 1) return false;
  if (fp->FReadEndian(&half_y_, sizeof(half_y_), 1) != 1) return false;
  no_ = ni_ * (2*half_x_ + 1) * (2*half_y_ + 1);
  return true;
}

Forward()

void tesseract::Convolve::Forward ( bool  debug, const NetworkIO &  input, const TransposedArray *  input_transpose, NetworkScratch *  scratch, NetworkIO *  output  )

virtual

Reimplemented from tesseract::Network.

Definition at line 54 of file convolve.cpp.

                                                                   {
  output->Resize(input, no_);
  int y_scale = 2 * half_y_ + 1;
  StrideMap::Index dest_index(output->stride_map());
  do {
    // Stack x_scale groups of y_scale * ni_ inputs together.
    int t = dest_index.t();
    int out_ix = 0;
    for (int x = -half_x_; x <= half_x_; ++x, out_ix += y_scale * ni_) {
      StrideMap::Index x_index(dest_index);
      if (!x_index.AddOffset(x, FD_WIDTH)) {
        // This x is outside the image.
        output->Randomize(t, out_ix, y_scale * ni_, randomizer_);
      } else {
        int out_iy = out_ix;
        for (int y = -half_y_; y <= half_y_; ++y, out_iy += ni_) {
          StrideMap::Index y_index(x_index);
          if (!y_index.AddOffset(y, FD_HEIGHT)) {
            // This y is outside the image.
            output->Randomize(t, out_iy, ni_, randomizer_);
          } else {
            output->CopyTimeStepGeneral(t, out_iy, ni_, input, y_index.t(), 0);
          }
        }
      }
    }
  } while (dest_index.Increment());
  if (debug) DisplayForward(*output);
}

Serialize()

bool tesseract::Convolve::Serialize ( TFile *  fp ) const

virtual

Reimplemented from tesseract::Network.

Definition at line 37 of file convolve.cpp.

                                        {
  if (!Network::Serialize(fp)) return false;
  if (fp->FWrite(&half_x_, sizeof(half_x_), 1) != 1) return false;
  if (fp->FWrite(&half_y_, sizeof(half_y_), 1) != 1) return false;
  return true;
}

spec()

virtual STRING tesseract::Convolve::spec ( ) const

inlinevirtual

Reimplemented from tesseract::Network.

Definition at line 40 of file convolve.h.

                              {
    STRING spec;
    spec.add_str_int("C", half_x_ * 2 + 1);
    spec.add_str_int(",", half_y_ * 2 + 1);
    return spec;
  }

Member Data Documentation

half_x_

inT32 tesseract::Convolve::half_x_

protected

Definition at line 66 of file convolve.h.

half_y_

inT32 tesseract::Convolve::half_y_

protected

Definition at line 67 of file convolve.h.

---

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

• /home/stefan/src/github/tesseract-ocr/tesseract/lstm/convolve.h
• /home/stefan/src/github/tesseract-ocr/tesseract/lstm/convolve.cpp