提交 38f472e4 编写于 作者: C chenguowei01

normal initialize and resize bilinear

上级 351d3ae0
...@@ -19,6 +19,7 @@ import paddle.fluid as fluid ...@@ -19,6 +19,7 @@ import paddle.fluid as fluid
from paddle.fluid.param_attr import ParamAttr from paddle.fluid.param_attr import ParamAttr
from paddle.fluid.layer_helper import LayerHelper from paddle.fluid.layer_helper import LayerHelper
from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear
from paddle.fluid.initializer import Normal
try: try:
from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
except: except:
...@@ -140,7 +141,8 @@ class HRNet(fluid.dygraph.Layer): ...@@ -140,7 +141,8 @@ class HRNet(fluid.dygraph.Layer):
filter_size=1, filter_size=1,
stride=1, stride=1,
padding=0, padding=0,
param_attr=ParamAttr(name='conv-1_weights')) param_attr=ParamAttr(
initializer=Normal(scale=0.001), name='conv-1_weights'))
def forward(self, x, label=None, mode='train'): def forward(self, x, label=None, mode='train'):
input_shape = x.shape[2:] input_shape = x.shape[2:]
...@@ -218,7 +220,8 @@ class ConvBNLayer(fluid.dygraph.Layer): ...@@ -218,7 +220,8 @@ class ConvBNLayer(fluid.dygraph.Layer):
padding=(filter_size - 1) // 2, padding=(filter_size - 1) // 2,
groups=groups, groups=groups,
act=None, act=None,
param_attr=ParamAttr(name=name + "_weights"), param_attr=ParamAttr(
initializer=Normal(scale=0.001), name=name + "_weights"),
bias_attr=False) bias_attr=False)
bn_name = name + '_bn' bn_name = name + '_bn'
self._batch_norm = BatchNorm( self._batch_norm = BatchNorm(
...@@ -646,7 +649,7 @@ class FuseLayers(fluid.dygraph.Layer): ...@@ -646,7 +649,7 @@ class FuseLayers(fluid.dygraph.Layer):
y = self.residual_func_list[residual_func_idx](input[j]) y = self.residual_func_list[residual_func_idx](input[j])
residual_func_idx += 1 residual_func_idx += 1
y = fluid.layers.resize_nearest(input=y, scale=2**(j - i)) y = fluid.layers.resize_bilinear(input=y, scale=2**(j - i))
residual = fluid.layers.elementwise_add( residual = fluid.layers.elementwise_add(
x=residual, y=y, act=None) x=residual, y=y, act=None)
elif j < i: elif j < i:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册