#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_t flags)

virtual int	InitWeights (float range, TRand *randomizer)

virtual int	RemapOutputs (int old_no, const std::vector< int > &code_map)

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, float adam_beta, 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_t	half_x_

int32_t	half_y_

Protected Attributes inherited from tesseract::Network
NetworkType	type_

TrainingState	training_

bool	needs_to_backprop_

int32_t	network_flags_

int32_t	ni_

int32_t	no_

int32_t	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.

33 {

34 }

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->CopyAll(*delta_sum);
   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_t tesseract::Convolve::half_x_

protected

Definition at line 66 of file convolve.h.

◆ half_y_

int32_t 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/stweil/src/github/tesseract-ocr/tesseract/lstm/convolve.h
/home/stweil/src/github/tesseract-ocr/tesseract/lstm/convolve.cpp

Public Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ Convolve()

◆ ~Convolve()

Member Function Documentation

◆ Backward()

◆ DeSerialize()

◆ Forward()

◆ Serialize()

◆ spec()

Member Data Documentation

◆ half_x_

◆ half_y_