#include <input.h>

Inheritance diagram for tesseract::Input:

Public Member Functions
	Input (const STRING &name, int ni, int no)

	Input (const STRING &name, const StaticShape &shape)

virtual	~Input ()=default

STRING	spec () const override

StaticShape	InputShape () const override

StaticShape	OutputShape (const StaticShape &input_shape) const override

bool	Serialize (TFile *fp) const override

bool	DeSerialize (TFile *fp) override

int	XScaleFactor () const override

void	CacheXScaleFactor (int factor) 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

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	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)

Static Public Member Functions
static Pix *	PrepareLSTMInputs (const ImageData &image_data, const Network network, int min_width, TRand randomizer, float *image_scale)

static void	PreparePixInput (const StaticShape &shape, const Pix pix, TRand randomizer, NetworkIO *input)

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)

Additional Inherited Members
Protected Member Functions inherited from tesseract::Network
double	Random (double range)

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_

Static Protected Attributes inherited from tesseract::Network
static char const *const	kTypeNames [NT_COUNT]

Detailed Description

Definition at line 28 of file input.h.

Constructor & Destructor Documentation

◆ Input() [1/2]

tesseract::Input::Input	(	const STRING &	name,
		int	ni,
		int	no
	)

Definition at line 31 of file input.cpp.

32 : Network(NT_INPUT, name, ni, no), cached_x_scale_(1) {}

tesseract::Network::Network

Network()

Definition: network.cpp:76

tesseract::Network::name

const STRING & name() const

Definition: network.h:138

tesseract::NT_INPUT

Definition: network.h:45

◆ Input() [2/2]

tesseract::Input::Input	(	const STRING &	name,
		const StaticShape &	shape
	)

Definition at line 33 of file input.cpp.

     : Network(NT_INPUT, name, shape.height(), shape.depth()),
       shape_(shape),
       cached_x_scale_(1) {
   if (shape.height() == 1) ni_ = shape.depth();
 }

◆ ~Input()

virtual tesseract::Input::~Input ( )

virtualdefault

Member Function Documentation

◆ Backward()

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

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 73 of file input.cpp.

                                              {
   tprintf("Input::Backward should not be called!!\n");
   return false;
 }

◆ CacheXScaleFactor()

void tesseract::Input::CacheXScaleFactor ( int factor )

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 59 of file input.cpp.

                                         {
   cached_x_scale_ = factor;
 }

◆ DeSerialize()

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

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 46 of file input.cpp.

                                  {
   return shape_.DeSerialize(fp);
 }

◆ Forward()

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

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 65 of file input.cpp.

                                                                 {
   *output = input;
 }

◆ InputShape()

StaticShape tesseract::Input::InputShape ( ) const

inlineoverridevirtual

Reimplemented from tesseract::Network.

Definition at line 44 of file input.h.

44 { return shape_; }

◆ OutputShape()

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

inlineoverridevirtual

Reimplemented from tesseract::Network.

Definition at line 47 of file input.h.

                                                                          {
     return shape_;
   }

◆ PrepareLSTMInputs()

Pix * tesseract::Input::PrepareLSTMInputs	(	const ImageData &	image_data,
		const Network *	network,
		int	min_width,
		TRand *	randomizer,
		float *	image_scale
	)

static

Definition at line 84 of file input.cpp.

                                                                      {
   // Note that NumInputs() is defined as input image height.
   int target_height = network->NumInputs();
   int width, height;
   Pix* pix = image_data.PreScale(target_height, kMaxInputHeight, image_scale,
                                  &width, &height, nullptr);
   if (pix == nullptr) {
     tprintf("Bad pix from ImageData!\n");
     return nullptr;
   }
   if (width <= min_width || height < min_width) {
     tprintf("Image too small to scale!! (%dx%d vs min width of %d)\n", width,
             height, min_width);
     pixDestroy(&pix);
     return nullptr;
   }
   return pix;
 }

◆ PreparePixInput()

void tesseract::Input::PreparePixInput	(	const StaticShape &	shape,
		const Pix *	pix,
		TRand *	randomizer,
		NetworkIO *	input
	)

static

Definition at line 112 of file input.cpp.

                                                                  {
   bool color = shape.depth() == 3;
   Pix* var_pix = const_cast<Pix*>(pix);
   int depth = pixGetDepth(var_pix);
   Pix* normed_pix = nullptr;
   // On input to BaseAPI, an image is forced to be 1, 8 or 24 bit, without
   // colormap, so we just have to deal with depth conversion here.
   if (color) {
     // Force RGB.
     if (depth == 32)
       normed_pix = pixClone(var_pix);
     else
       normed_pix = pixConvertTo32(var_pix);
   } else {
     // Convert non-8-bit images to 8 bit.
     if (depth == 8)
       normed_pix = pixClone(var_pix);
     else
       normed_pix = pixConvertTo8(var_pix, false);
   }
   int height = pixGetHeight(normed_pix);
   int target_height = shape.height();
   if (target_height == 1) target_height = shape.depth();
   if (target_height != 0 && target_height != height) {
     // Get the scaled image.
     float im_factor = static_cast<float>(target_height) / height;
     Pix* scaled_pix = pixScale(normed_pix, im_factor, im_factor);
     pixDestroy(&normed_pix);
     normed_pix = scaled_pix;
   }
   input->FromPix(shape, normed_pix, randomizer);
   pixDestroy(&normed_pix);
 }

◆ Serialize()

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

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 41 of file input.cpp.

                                      {
   return Network::Serialize(fp) && shape_.Serialize(fp);
 }

◆ spec()

STRING tesseract::Input::spec ( ) const

inlineoverridevirtual

Reimplemented from tesseract::Network.

Definition at line 34 of file input.h.

                                {
     STRING spec;
     spec.add_str_int("", shape_.batch());
     spec.add_str_int(",", shape_.height());
     spec.add_str_int(",", shape_.width());
     spec.add_str_int(",", shape_.depth());
     return spec;
   }

◆ XScaleFactor()

int tesseract::Input::XScaleFactor ( ) const

overridevirtual

Reimplemented from tesseract::Network.

Definition at line 53 of file input.cpp.

                               {
   return 1;
 }

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

/usr/src/tesseract-ocr.master/src/lstm/input.h
/usr/src/tesseract-ocr.master/src/lstm/input.cpp

Public Member Functions

Static Public Member Functions

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ Input() [1/2]

◆ Input() [2/2]

◆ ~Input()

Member Function Documentation

◆ Backward()

◆ CacheXScaleFactor()

◆ DeSerialize()

◆ Forward()

◆ InputShape()

◆ OutputShape()

◆ PrepareLSTMInputs()

◆ PreparePixInput()

◆ Serialize()

◆ spec()

◆ XScaleFactor()