#include <reconfig.h>

Inheritance diagram for tesseract::Reconfig:

Public Member Functions
TESS_API	Reconfig (const char *name, int ni, int x_scale, int y_scale)

	~Reconfig () override=default

StaticShape	OutputShape (const StaticShape &input_shape) const override

std::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 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
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_

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 reconfig.h.

Constructor & Destructor Documentation

◆ Reconfig()

tesseract::Reconfig::Reconfig	(	const char *	name,
		int	ni,
		int	x_scale,
		int	y_scale
	)

Definition at line 23 of file reconfig.cpp.

    : Network(NT_RECONFIG, name, ni, ni * x_scale * y_scale)
    , x_scale_(x_scale)
    , y_scale_(y_scale) {}

◆ ~Reconfig()

tesseract::Reconfig::~Reconfig ( )

overridedefault

Member Function Documentation

◆ Backward()

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

overridevirtual

Implements tesseract::Network.

Definition at line 93 of file reconfig.cpp.

                                                {
  back_deltas->ResizeToMap(fwd_deltas.int_mode(), back_map_, ni_);
  StrideMap::Index src_index(fwd_deltas.stride_map());
  do {
    int in_t = src_index.t();
    StrideMap::Index dest_index(back_deltas->stride_map(), src_index.index(FD_BATCH),
                                src_index.index(FD_HEIGHT) * y_scale_,
                                src_index.index(FD_WIDTH) * x_scale_);
    // Unstack x_scale_ groups of y_scale_ inputs that are together.
    for (int x = 0; x < x_scale_; ++x) {
      for (int y = 0; y < y_scale_; ++y) {
        StrideMap::Index dest_xy(dest_index);
        if (dest_xy.AddOffset(x, FD_WIDTH) && dest_xy.AddOffset(y, FD_HEIGHT)) {
          back_deltas->CopyTimeStepGeneral(dest_xy.t(), 0, ni_, fwd_deltas, in_t,
                                           (x * y_scale_ + y) * ni_);
        }
      }
    }
  } while (src_index.Increment());
  return needs_to_backprop_;
}

◆ DeSerialize()

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

overridevirtual

Implements tesseract::Network.

Definition at line 56 of file reconfig.cpp.

                                    {
  if (!fp->DeSerialize(&x_scale_)) {
    return false;
  }
  if (!fp->DeSerialize(&y_scale_)) {
    return false;
  }
  no_ = ni_ * x_scale_ * y_scale_;
  return true;
}

◆ Forward()

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

overridevirtual

Implements tesseract::Network.

Definition at line 69 of file reconfig.cpp.

                                                                   {
  output->ResizeScaled(input, x_scale_, y_scale_, no_);
  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_);
    // Stack x_scale_ groups of y_scale_ inputs together.
    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->CopyTimeStepGeneral(out_t, (x * y_scale_ + y) * ni_, ni_, input, src_xy.t(), 0);
        }
      }
    }
  } while (dest_index.Increment());
}

◆ OutputShape()

StaticShape tesseract::Reconfig::OutputShape ( const StaticShape & input_shape ) const

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 30 of file reconfig.cpp.

                                                                      {
  StaticShape result = input_shape;
  result.set_height(result.height() / y_scale_);
  result.set_width(result.width() / x_scale_);
  if (type_ != NT_MAXPOOL) {
    result.set_depth(result.depth() * y_scale_ * x_scale_);
  }
  return result;
}

◆ Serialize()

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

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 51 of file reconfig.cpp.

                                        {
  return Network::Serialize(fp) && fp->Serialize(&x_scale_) && fp->Serialize(&y_scale_);
}

◆ spec()

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

inlineoverridevirtual

Reimplemented from tesseract::Network.

Definition at line 41 of file reconfig.h.

                                  {
    return "S" + std::to_string(y_scale_) + "," + std::to_string(x_scale_);
  }

◆ XScaleFactor()

int tesseract::Reconfig::XScaleFactor ( ) const

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 46 of file reconfig.cpp.

                                 {
  return x_scale_;
}

Member Data Documentation

◆ back_map_

StrideMap tesseract::Reconfig::back_map_

protected

Definition at line 75 of file reconfig.h.

◆ x_scale_

int32_t tesseract::Reconfig::x_scale_

protected

Definition at line 77 of file reconfig.h.

◆ y_scale_

int32_t tesseract::Reconfig::y_scale_

protected

Definition at line 78 of file reconfig.h.

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

/media/home/debian/src/github/tesseract-ocr/tesseract/src/lstm/reconfig.h
/media/home/debian/src/github/tesseract-ocr/tesseract/src/lstm/reconfig.cpp

Public Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ Reconfig()

◆ ~Reconfig()

Member Function Documentation

◆ Backward()

◆ DeSerialize()

◆ Forward()

◆ OutputShape()

◆ Serialize()

◆ spec()

◆ XScaleFactor()

Member Data Documentation

◆ back_map_

◆ x_scale_

◆ y_scale_