Add pad_h, pad_w, kernel_size_h, kernel_size_w, stride_h, stride_w to support pooling on rectangle regions.
This commit is contained in:
Ronghang Hu 2014-07-03 20:30:25 -07:00
Родитель 9dc8734600
Коммит 4e5ef959ab
1 изменённых файлов: 11 добавлений и 5 удалений

Просмотреть файл

@ -405,11 +405,17 @@ message PoolingParameter {
STOCHASTIC = 2;
}
optional PoolMethod pool = 1 [default = MAX]; // The pooling method
optional uint32 kernel_size = 2; // The kernel size
optional uint32 stride = 3 [default = 1]; // The stride
// The padding size -- currently implemented only for average and max pooling.
// average pooling zero pads. max pooling -inf pads.
optional uint32 pad = 4 [default = 0];
// Pad, kernel size, and stride are all given as a single value for equal
// dimensions in height and width or as Y, X pairs.
optional uint32 pad = 4 [default = 0]; // The padding size (equal in Y, X)
optional uint32 pad_h = 9 [default = 0]; // The padding height
optional uint32 pad_w = 10 [default = 0]; // The padding width
optional uint32 kernel_size = 2; // The kernel size (square)
optional uint32 kernel_size_h = 5; // The kernel height
optional uint32 kernel_size_w = 6; // The kernel width
optional uint32 stride = 3 [default = 1]; // The stride (equal in Y, X)
optional uint32 stride_h = 7; // The stride height
optional uint32 stride_w = 8; // The stride width
}
// Message that stores parameters used by PowerLayer