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 158 of file keras_model.h.

Constructor & Destructor Documentation

keras::LayerMaxPooling::LayerMaxPooling ( )

inline

Definition at line 160 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 149 of file keras_model.cc.

References d, keras::DataChunk::get_3d(), util::size(), util::values(), 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 169 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 165 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 140 of file keras_model.h.

{ return m_name; }

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

inlinevirtual

Implements keras::Layer.

Definition at line 170 of file keras_model.h.

{ return 0; }

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

virtual

Implements keras::Layer.

Definition at line 92 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 141 of file keras_model.h.

int keras::LayerMaxPooling::m_pool_x

Definition at line 172 of file keras_model.h.

int keras::LayerMaxPooling::m_pool_y

Definition at line 173 of file keras_model.h.

---

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

• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Keras/keras_model.h
• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Keras/keras_model.cc