keras::KerasModel Class Reference

#include "keras_model.h"

Public Member Functions
	KerasModel (const std::string &input_fname)
	~KerasModel ()
std::vector< float >	compute_output (keras::DataChunk *dc)
unsigned int	get_input_rows () const
unsigned int	get_input_cols () const
int	get_output_length () const

Private Member Functions
void	load_weights (const std::string &input_fname)

Private Attributes
int	m_layers_cnt
std::vector< Layer * >	m_layers

Detailed Description

Definition at line 214 of file keras_model.h.

Constructor & Destructor Documentation

keras::KerasModel::KerasModel

(

const std::string &

input_fname

)

Definition at line 119 of file keras_model.cc.

                                                     {
  load_weights(input_fname);
}

keras::KerasModel::~KerasModel

(

)

Definition at line 472 of file keras_model.cc.

                             {
  for(int i = 0; i < (int)m_layers.size(); ++i) {
    delete m_layers[i];
  }
}

Member Function Documentation

std::vector< float > keras::KerasModel::compute_output

(

keras::DataChunk *

dc

)

Definition at line 403 of file keras_model.cc.

References keras::DataChunk::get_1d().

Referenced by main(), and nnet::KerasModelInterface::Run().

                                                                   {
  //cout << endl << "KerasModel compute output" << endl;
  //cout << "Input data size:" << endl;
  //dc->show_name();

  keras::DataChunk *inp = dc;
  keras::DataChunk *out = 0;
  for(int l = 0; l < (int)m_layers.size(); ++l) {
    //cout << "Processing layer " << m_layers[l]->get_name() << endl;
    out = m_layers[l]->compute_output(inp);

    //cout << "Input" << endl;
    //inp->show_name();
    //cout << "Output" << endl;
    //out->show_name();

    if (inp != dc) delete inp;
    inp = 0L;
    inp = out;
  }

  //cout << "Output: ";
  //out->show_values();

  std::vector<float> flat_out = out->get_1d();
  delete out;

  return flat_out;
}

unsigned int keras::KerasModel::get_input_cols ( ) const

inline

Definition at line 221 of file keras_model.h.

{ return m_layers.front()->get_input_cols(); }

unsigned int keras::KerasModel::get_input_rows ( ) const

inline

Definition at line 220 of file keras_model.h.

{ return m_layers.front()->get_input_rows(); }

int keras::KerasModel::get_output_length

(

)

const

Definition at line 478 of file keras_model.cc.

{
  int i = m_layers.size() - 1;
  while ((i > 0) && (m_layers[i]->get_output_units() == 0)) --i;
  return m_layers[i]->get_output_units();
}

void keras::KerasModel::load_weights ( const std::string &  input_fname )

private

Definition at line 433 of file keras_model.cc.

References fin, and keras::Layer::load_weights().

                                                            {
  cout << "Reading model from " << input_fname << endl;
  ifstream fin(input_fname.c_str());
  string layer_type = "";
  string tmp_str = "";
  int tmp_int = 0;

  fin >> tmp_str >> m_layers_cnt;
  cout << "Layers " << m_layers_cnt << endl;

  for(int layer = 0; layer < m_layers_cnt; ++layer) { // iterate over layers
    fin >> tmp_str >> tmp_int >> layer_type;
    cout << "Layer " << tmp_int << " " << layer_type << endl;

    Layer *l = 0L;
    if(layer_type == "Convolution2D") {
      l = new LayerConv2D();
    } else if(layer_type == "Activation") {
      l = new LayerActivation();
    } else if(layer_type == "MaxPooling2D") {
      l = new LayerMaxPooling();
    } else if(layer_type == "Flatten") {
      l = new LayerFlatten();
    } else if(layer_type == "Dense") {
      l = new LayerDense();
    } else if(layer_type == "Dropout") {
      continue; // we dont need dropout layer in prediciton mode
    }
    if(l == 0L) {
      cout << "Layer is empty, maybe it is not defined? Cannot define network." << endl;
      return;
    }
    l->load_weights(fin);
    m_layers.push_back(l);
  }

  fin.close();
}

Member Data Documentation

std::vector<Layer *> keras::KerasModel::m_layers

private

Definition at line 228 of file keras_model.h.

int keras::KerasModel::m_layers_cnt

private

Definition at line 227 of file keras_model.h.

---

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

• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Keras/keras_model.h
• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Keras/keras_model.cc