Add the pooling layer ceil_mode patch for DenseNet.

Author: vera121

include/caffe/layers/pooling_layer.hpp

@@ -54,6 +54,7 @@ class PoolingLayer : public Layer<Dtype> {
   Blob<Dtype> rand_idx_;
   Blob<int> max_idx_;
   int pad_type_; //CUSTOMIZATION
+  bool ceil_mode_;
 };
 
 }  // namespace caffe

src/caffe/layers/pooling_layer.cpp

@@ -35,6 +35,7 @@ void PoolingLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype>*>& bottom,
       || (!pool_param.has_stride_h() && !pool_param.has_stride_w()))
       << "Stride is stride OR stride_h and stride_w are required.";
   global_pooling_ = pool_param.global_pooling();
+  ceil_mode_ = pool_param.ceil_mode();
   if (global_pooling_) {
     kernel_h_ = bottom[0]->height();
     kernel_w_ = bottom[0]->width();
@@ -102,10 +103,19 @@ void PoolingLayer<Dtype>::Reshape(const vector<Blob<Dtype>*>& bottom,
   //<--CUSTOMIZATION
   switch (pad_type_) {
     case 0:
-      pooled_height_ = static_cast<int>(ceil(static_cast<float>(
-        height_ + 2 * pad_h_ - kernel_h_) / stride_h_)) + 1;
-      pooled_width_ = static_cast<int>(ceil(static_cast<float>(
-        width_ + 2 * pad_w_ - kernel_w_) / stride_w_)) + 1;
+      // Specify the structure by ceil or floor mode
+      if (ceil_mode_) {
+        pooled_height_ = static_cast<int>(ceil(static_cast<float>(
+          height_ + 2 * pad_h_ - kernel_h_) / stride_h_)) + 1;
+        pooled_width_ = static_cast<int>(ceil(static_cast<float>(
+          width_ + 2 * pad_w_ - kernel_w_) / stride_w_)) + 1;
+      }
+      else{
+        pooled_height_ = static_cast<int>(floor(static_cast<float>(
+    	  height_ + 2 * pad_h_ - kernel_h_) / stride_h_)) + 1;
+    	pooled_width_ = static_cast<int>(floor(static_cast<float>(
+    	  width_ + 2 * pad_w_ - kernel_w_) / stride_w_)) + 1;
+      }
 	  break;
     case 1: //for "SAME"padding
       pooled_height_ = static_cast<int>(ceil(static_cast<float>(height_) / static_cast<float>(stride_h_)));

src/caffe/proto/caffe.proto

@@ -1390,6 +1390,7 @@ message PoolingParameter {
   // If global_pooling then it will pool over the size of the bottom by doing
   // kernel_h = bottom->height and kernel_w = bottom->width
   optional bool global_pooling = 12 [default = false];
+  optional bool ceil_mode = 14 [default = true]; // Specify floor/ceil mode
 }
 
 message PowerParameter {