tesseract::NetworkBuilder Class Reference

#include <networkbuilder.h>

Public Member Functions
	NetworkBuilder (int num_softmax_outputs)
Network *	BuildFromString (const StaticShape &input_shape, char **str)

Static Public Member Functions
static bool	InitNetwork (int num_outputs, STRING network_spec, int append_index, int net_flags, float weight_range, TRand *randomizer, Network **network)

Detailed Description

Definition at line 36 of file networkbuilder.h.

Constructor & Destructor Documentation

NetworkBuilder()

tesseract::NetworkBuilder::NetworkBuilder ( int  num_softmax_outputs )

inlineexplicit

Definition at line 38 of file networkbuilder.h.

      : num_softmax_outputs_(num_softmax_outputs) {}

Member Function Documentation

BuildFromString()

Network * tesseract::NetworkBuilder::BuildFromString	(	const StaticShape &	input_shape,
		char **	str
	)

Definition at line 86 of file networkbuilder.cpp.

                                                     {
  SkipWhitespace(str);
  char code_ch = **str;
  if (code_ch == '[') {
    return ParseSeries(input_shape, nullptr, str);
  }
  if (input_shape.depth() == 0) {
    // There must be an input at this point.
    return ParseInput(str);
  }
  switch (code_ch) {
    case '(':
      return ParseParallel(input_shape, str);
    case 'R':
      return ParseR(input_shape, str);
    case 'S':
      return ParseS(input_shape, str);
    case 'C':
      return ParseC(input_shape, str);
    case 'M':
      return ParseM(input_shape, str);
    case 'L':
      return ParseLSTM(input_shape, str);
    case 'F':
      return ParseFullyConnected(input_shape, str);
    case 'O':
      return ParseOutput(input_shape, str);
    default:
      tprintf("Invalid network spec:%s\n", *str);
      return nullptr;
  }
  return nullptr;
}

InitNetwork()

bool tesseract::NetworkBuilder::InitNetwork ( int  num_outputs, STRING  network_spec, int  append_index, int  net_flags, float  weight_range, TRand *  randomizer, Network **  network  )

static

Definition at line 45 of file networkbuilder.cpp.

                                                    {
  NetworkBuilder builder(num_outputs);
  Series* bottom_series = nullptr;
  StaticShape input_shape;
  if (append_index >= 0) {
    // Split the current network after the given append_index.
    ASSERT_HOST(*network != nullptr && (*network)->type() == NT_SERIES);
    auto* series = static_cast<Series*>(*network);
    Series* top_series = nullptr;
    series->SplitAt(append_index, &bottom_series, &top_series);
    if (bottom_series == nullptr || top_series == nullptr) {
      tprintf("Yikes! Splitting current network failed!!\n");
      return false;
    }
    input_shape = bottom_series->OutputShape(input_shape);
    delete top_series;
  }
  char* str_ptr = &network_spec[0];
  *network = builder.BuildFromString(input_shape, &str_ptr);
  if (*network == nullptr) return false;
  (*network)->SetNetworkFlags(net_flags);
  (*network)->InitWeights(weight_range, randomizer);
  (*network)->SetupNeedsBackprop(false);
  if (bottom_series != nullptr) {
    bottom_series->AppendSeries(*network);
    *network = bottom_series;
  }
  (*network)->CacheXScaleFactor((*network)->XScaleFactor());
  return true;
}

---

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

• /tesseract/src/training/networkbuilder.h
• /tesseract/src/training/networkbuilder.cpp