tesseract::Maxpool Class Reference

#include <maxpool.h>

Inheritance diagram for tesseract::Maxpool:

Public Member Functions
	Maxpool (const STRING &name, int ni, int x_scale, int y_scale)
	~Maxpool () override=default
STRING	spec () 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::Reconfig
	Reconfig (const STRING &name, int ni, int x_scale, int y_scale)
	~Reconfig () override=default
StaticShape	OutputShape (const StaticShape &input_shape) const override
STRING	spec () const override
int	XScaleFactor () 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 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
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 void	CacheXScaleFactor (int factor)
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)

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)
Protected Attributes inherited from tesseract::Reconfig
StrideMap	back_map_
int32_t	x_scale_
int32_t	y_scale_
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_

Detailed Description

Definition at line 29 of file maxpool.h.

Constructor & Destructor Documentation

Maxpool()

tesseract::Maxpool::Maxpool	(	const STRING &	name,
		int	ni,
		int	x_scale,
		int	y_scale
	)

Definition at line 22 of file maxpool.cpp.

  : Reconfig(name, ni, x_scale, y_scale) {
  type_ = NT_MAXPOOL;
  no_ = ni;
}

~Maxpool()

tesseract::Maxpool::~Maxpool ( )

overridedefault

Member Function Documentation

Backward()

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

overridevirtual

Implements tesseract::Network.

Definition at line 71 of file maxpool.cpp.

                                               {
  back_deltas->ResizeToMap(fwd_deltas.int_mode(), back_map_, ni_);
  back_deltas->MaxpoolBackward(fwd_deltas, maxes_);
  return true;
}

DeSerialize()

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

overridevirtual

Implements tesseract::Network.

Definition at line 29 of file maxpool.cpp.

                                   {
  bool result = Reconfig::DeSerialize(fp);
  no_ = ni_;
  return result;
}

Forward()

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

overridevirtual

Implements tesseract::Network.

Definition at line 37 of file maxpool.cpp.

                                                                  {
  output->ResizeScaled(input, x_scale_, y_scale_, no_);
  maxes_.ResizeNoInit(output->Width(), ni_);
  back_map_ = input.stride_map();
 
  StrideMap::Index dest_index(output->stride_map());
  do {
    int out_t = dest_index.t();
    StrideMap::Index src_index(input.stride_map(), dest_index.index(FD_BATCH),
                               dest_index.index(FD_HEIGHT) * y_scale_,
                               dest_index.index(FD_WIDTH) * x_scale_);
    // Find the max input out of x_scale_ groups of y_scale_ inputs.
    // Do it independently for each input dimension.
    int* max_line = maxes_[out_t];
    int in_t = src_index.t();
    output->CopyTimeStepFrom(out_t, input, in_t);
    for (int i = 0; i < ni_; ++i) {
      max_line[i] = in_t;
    }
    for (int x = 0; x < x_scale_; ++x) {
      for (int y = 0; y < y_scale_; ++y) {
        StrideMap::Index src_xy(src_index);
        if (src_xy.AddOffset(x, FD_WIDTH) && src_xy.AddOffset(y, FD_HEIGHT)) {
          output->MaxpoolTimeStep(out_t, input, src_xy.t(), max_line);
        }
      }
    }
  } while (dest_index.Increment());
}

spec()

STRING tesseract::Maxpool::spec ( ) const

inlineoverridevirtual

Reimplemented from tesseract::Network.

Definition at line 35 of file maxpool.h.

                               {
    STRING spec;
    spec.add_str_int("Mp", y_scale_);
    spec.add_str_int(",", x_scale_);
    return spec;
  }

---

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

• /tesseract/src/lstm/maxpool.h
• /tesseract/src/lstm/maxpool.cpp