From 9905c20130783046248b8c17a3cbc17847019d22 Mon Sep 17 00:00:00 2001 From: Ronghang Hu Date: Thu, 3 Jul 2014 20:33:50 -0700 Subject: [PATCH] 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. --- include/caffe/vision_layers.hpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp index b68dcbf..62b65bc 100644 --- a/include/caffe/vision_layers.hpp +++ b/include/caffe/vision_layers.hpp @@ -361,9 +361,9 @@ class PoolingLayer : public Layer { const vector& propagate_down, vector*>* 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_; -- 2.7.4