提交 4ca9c395 编写于 作者: D dangqingqing

minor changes

上级 876fa193
......@@ -92,7 +92,7 @@ message PoolConfig {
optional uint32 start = 4;
// Defines the stride size between successive pooling squares.
required uint32 stride = 5;
required uint32 stride = 5 [default = 1];
// The size of output feature map.
required uint32 output_x = 6;
......@@ -105,19 +105,19 @@ message PoolConfig {
optional uint32 padding = 8 [default = 0];
// if not set, use size_x
optional uint32 size_y = 9 [default = 0];
optional uint32 size_y = 9;
// if not set, use stride
optional uint32 stride_y = 10 [default = 0];
optional uint32 stride_y = 10;
// if not set, use output_x
optional uint32 output_y = 11 [default = 0];
optional uint32 output_y = 11;
// if not set, use img_size
optional uint32 img_size_y = 12 [default = 0];
optional uint32 img_size_y = 12;
// if not set, use padding
optional uint32 padding_y = 13 [default = 0];
optional uint32 padding_y = 13;
}
message SppConfig {
......
......@@ -815,9 +815,9 @@ class Pool(Cfg):
size_y=None,
img_width=None,
start=None,
stride=1,
stride=None, # 1 by defalut in protobuf
stride_y=None,
padding=0,
padding=None, # 0 by defalut in protobuf
padding_y=None):
self.add_keys(locals())
......@@ -1111,9 +1111,9 @@ def parse_pool(pool, input_layer_name, pool_conf):
config_assert(not pool.start, "start is deprecated in pooling.")
pool_conf.padding = pool.padding
if pool.padding is not None
pool_conf.padding = pool.padding
pool_conf.padding_y = default(pool.padding_y, pool_conf.padding)
pool_conf.output_x = cnn_output_size(
pool_conf.img_size, pool_conf.size_x, pool_conf.padding,
pool_conf.stride, False)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册