keras::LayerMaxPooling Class Reference

#include "keras_model.h"

Inheritance diagram for keras::LayerMaxPooling:

Public Member Functions
	LayerMaxPooling ()
void	load_weights (std::ifstream &fin)
keras::DataChunk *	compute_output (keras::DataChunk *)
virtual unsigned int	get_input_rows () const
virtual unsigned int	get_input_cols () const
virtual unsigned int	get_output_units () const
std::string	get_name ()

Public Attributes
int	m_pool_x
int	m_pool_y
std::string	m_name

Detailed Description

Definition at line 148 of file keras_model.h.

Constructor & Destructor Documentation

keras::LayerMaxPooling::LayerMaxPooling ( )

inline

Definition at line 150 of file keras_model.h.

References fin.

: Layer("MaxPooling2D") {};

Member Function Documentation

keras::DataChunk * keras::LayerMaxPooling::compute_output ( keras::DataChunk *  dc )

virtual

Implements keras::Layer.

Definition at line 145 of file keras_model.cc.

References d, keras::DataChunk::get_3d(), x, and y.

                                                                       {
  vector<vector<vector<float> > > im = dc->get_3d();
  vector<vector<vector<float> > > y_ret;
  for(unsigned int i = 0; i < im.size(); ++i) {
    vector<vector<float> > tmp_y;
    for(unsigned int j = 0; j < (unsigned int)(im[0].size()/m_pool_x); ++j) {
      tmp_y.push_back(vector<float>((int)(im[0][0].size()/m_pool_y), 0.0));
    }
    y_ret.push_back(tmp_y);
  }
  for(unsigned int d = 0; d < y_ret.size(); ++d) {
    for(unsigned int x = 0; x < y_ret[0].size(); ++x) {
      unsigned int start_x = x*m_pool_x;
      unsigned int end_x = start_x + m_pool_x;
      for(unsigned int y = 0; y < y_ret[0][0].size(); ++y) {
        unsigned int start_y = y*m_pool_y;
        unsigned int end_y = start_y + m_pool_y;

        vector<float> values;
        for(unsigned int i = start_x; i < end_x; ++i) {
          for(unsigned int j = start_y; j < end_y; ++j) {
            values.push_back(im[d][i][j]);
          }
        }
        y_ret[d][x][y] = *max_element(values.begin(), values.end());
      }
    }
  }
  keras::DataChunk *out = new keras::DataChunk2D();
  out->set_data(y_ret);
  return out;
}

virtual unsigned int keras::LayerMaxPooling::get_input_cols ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 156 of file keras_model.h.

{ return 0; } // same as for rows

virtual unsigned int keras::LayerMaxPooling::get_input_rows ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 155 of file keras_model.h.

{ return 0; } // look for the value in the preceding layer

std::string keras::Layer::get_name ( )

inlineinherited

Definition at line 131 of file keras_model.h.

{ return m_name; }

virtual unsigned int keras::LayerMaxPooling::get_output_units ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 157 of file keras_model.h.

{ return 0; }

void keras::LayerMaxPooling::load_weights ( std::ifstream &  fin )

virtual

Implements keras::Layer.

Definition at line 89 of file keras_model.cc.

                                                        {
  fin >> m_pool_x >> m_pool_y;
  cout << "MaxPooling " << m_pool_x << "x" << m_pool_y << endl;
}

Member Data Documentation

std::string keras::Layer::m_name

inherited

Definition at line 132 of file keras_model.h.

int keras::LayerMaxPooling::m_pool_x

Definition at line 159 of file keras_model.h.

int keras::LayerMaxPooling::m_pool_y

Definition at line 160 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