Update vision_layers.hpp

Replace pad_, kernel_size_, stride_ with pad_h_, pad_w_, kernel_size_h_, kernel_size_w_, stride_h_, stride_w_ to support pooling on rectangle regions.

diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp
index b68dcbf..62b65bc 100644 (file)
--- a/include/caffe/vision_layers.hpp
+++ b/include/caffe/vision_layers.hpp
@@ -361,9 +361,9 @@ class PoolingLayer : public Layer<Dtype> {
       const vector<bool>& propagate_down, vector<Blob<Dtype>*>* bottom);
 
   int max_top_blobs_;
-  int kernel_size_;
-  int stride_;
-  int pad_;
+  int kernel_size_h_, kernel_size_w_;
+  int stride_h_, stride_w_;
+  int pad_h_, pad_w_;
   int channels_;
   int height_;
   int width_;