@@ -1827,7 +1827,6 @@ def img_pool_layer(input, pool_size, name=None,
1827
1827
@layer_support ()
1828
1828
def spp_layer (input , name = None , num_channels = None , pool_type = None ,
1829
1829
pyramid_height = None , img_width = None , layer_attr = None ):
1830
- pass
1831
1830
"""
1832
1831
Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition.
1833
1832
The details please refer to
@@ -1864,7 +1863,7 @@ def spp_layer(input, name=None, num_channels=None, pool_type=None,
1864
1863
if (isinstance (pool_type , AvgPooling ) or isinstance (pool_type , MaxPooling )):
1865
1864
type_name += '-projection'
1866
1865
1867
- Layer (
1866
+ l = Layer (
1868
1867
name = name ,
1869
1868
type = LayerType .SPP_LAYER ,
1870
1869
inputs = Input (input .name ,
@@ -1875,8 +1874,8 @@ def spp_layer(input, name=None, num_channels=None, pool_type=None,
1875
1874
),
1876
1875
** ExtraLayerAttribute .to_kwargs (layer_attr )
1877
1876
)
1878
- return LayerOutput (name , LayerType .SPP_LAYER , parents = [input ],
1879
- num_filters = num_channels )
1877
+ return LayerOutput (name , layer_type = LayerType .SPP_LAYER , parents = [input ],
1878
+ num_filters = num_channels , size = l . config . size )
1880
1879
1881
1880
1882
1881
def __img_norm_layer__ (name , input , size , norm_type , scale , power ,
0 commit comments