#include <convolve.h>

Inheritance diagram for tesseract::Convolve:

Public Member Functions
TESS_API	Convolve (const std::string &name, int ni, int half_x, int half_y)

	~Convolve () override=default

std::string	spec () const override

bool	Serialize (TFile *fp) const override

bool	DeSerialize (TFile *fp) override

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

bool	Backward (bool debug, const NetworkIO &fwd_deltas, NetworkScratch scratch, NetworkIO back_deltas) override

Public Member Functions inherited from tesseract::Network
	Network ()

	Network (NetworkType type, const std::string &name, int ni, int no)

virtual	~Network ()=default

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 std::string &	name () const

virtual std::string	spec () 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 ()=0

virtual bool	Serialize (TFile *fp) const

virtual bool	DeSerialize (TFile *fp)=0

virtual void	Update (float learning_rate, float momentum, float adam_beta, int num_samples)

virtual void	CountAlternators (const Network &other, TFloat same, TFloat changed) const

virtual void	Forward (bool debug, const NetworkIO &input, const TransposedArray input_transpose, NetworkScratch scratch, NetworkIO *output)=0

virtual bool	Backward (bool debug, const NetworkIO &fwd_deltas, NetworkScratch scratch, NetworkIO back_deltas)=0

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_

std::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 (Image pix, ScrollView *window)

Protected Member Functions inherited from tesseract::Network
TFloat	Random (TFloat range)

Detailed Description

Definition at line 31 of file convolve.h.

Constructor & Destructor Documentation

◆ Convolve()

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

Definition at line 31 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 ( )

overridedefault

Member Function Documentation

◆ Backward()

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

overridevirtual

Implements tesseract::Network.

Definition at line 92 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 )

overridevirtual

Implements tesseract::Network.

Definition at line 42 of file convolve.cpp.

                                    {
  if (!fp->DeSerialize(&half_x_)) {
    return false;
  }
  if (!fp->DeSerialize(&half_y_)) {
    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
	)

overridevirtual

Implements tesseract::Network.

Definition at line 55 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());
#ifndef GRAPHICS_DISABLED
  if (debug) {
    DisplayForward(*output);
  }
#endif
}

◆ Serialize()

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

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 37 of file convolve.cpp.

                                        {
  return Network::Serialize(fp) && fp->Serialize(&half_x_) && fp->Serialize(&half_y_);
}

◆ spec()

std::string tesseract::Convolve::spec ( ) const

inlineoverridevirtual

Reimplemented from tesseract::Network.

Definition at line 39 of file convolve.h.

                                  {
    return "C" + std::to_string(half_y_ * 2 + 1) + "," + std::to_string(half_x_ * 2 + 1);
  }

Member Data Documentation

◆ half_x_

int32_t tesseract::Convolve::half_x_

protected

Definition at line 65 of file convolve.h.

◆ half_y_

int32_t tesseract::Convolve::half_y_

protected

Definition at line 66 of file convolve.h.

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

/media/home/debian/src/github/tesseract-ocr/tesseract/src/lstm/convolve.h
/media/home/debian/src/github/tesseract-ocr/tesseract/src/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_