提交 ed9f214c 编写于 作者: C Channingss

FastSCNN: resize keep 'align_corners' mode

上级 80ba5a9b
...@@ -86,27 +86,26 @@ class FastSCNN(object): ...@@ -86,27 +86,26 @@ class FastSCNN(object):
64, 128, 128) 64, 128, 128)
with scope('classifier'): with scope('classifier'):
logit = self._classifier(x, 128) logit = self._classifier(x, 128)
logit = fluid.layers.resize_bilinear( logit = fluid.layers.resize_bilinear(logit, size, align_mode=0)
logit, size, align_corners=False, align_mode=1)
if len(self.multi_loss_weight) == 3: if len(self.multi_loss_weight) == 3:
with scope('aux_layer_higher'): with scope('aux_layer_higher'):
higher_logit = self._aux_layer(higher_res_features, higher_logit = self._aux_layer(higher_res_features,
self.num_classes) self.num_classes)
higher_logit = fluid.layers.resize_bilinear( higher_logit = fluid.layers.resize_bilinear(
higher_logit, size, align_corners=False, align_mode=1) higher_logit, size, align_mode=0)
with scope('aux_layer_lower'): with scope('aux_layer_lower'):
lower_logit = self._aux_layer(lower_res_feature, lower_logit = self._aux_layer(lower_res_feature,
self.num_classes) self.num_classes)
lower_logit = fluid.layers.resize_bilinear( lower_logit = fluid.layers.resize_bilinear(
lower_logit, size, align_corners=False, align_mode=1) lower_logit, size, align_mode=0)
logit = (logit, higher_logit, lower_logit) logit = (logit, higher_logit, lower_logit)
elif len(self.multi_loss_weight) == 2: elif len(self.multi_loss_weight) == 2:
with scope('aux_layer_higher'): with scope('aux_layer_higher'):
higher_logit = self._aux_layer(higher_res_features, higher_logit = self._aux_layer(higher_res_features,
self.num_classes) self.num_classes)
higher_logit = fluid.layers.resize_bilinear( higher_logit = fluid.layers.resize_bilinear(
higher_logit, size, align_corners=False, align_mode=1) higher_logit, size, align_mode=0)
logit = (logit, higher_logit) logit = (logit, higher_logit)
else: else:
logit = (logit, ) logit = (logit, )
...@@ -318,8 +317,7 @@ class FastSCNN(object): ...@@ -318,8 +317,7 @@ class FastSCNN(object):
data_bn, data_bn,
out_shape=fluid.layers.shape(input)[2:], out_shape=fluid.layers.shape(input)[2:],
name=psp_name + '_interp', name=psp_name + '_interp',
align_corners=False, align_mode=0)
align_mode=1)
cat_layers.append(interp) cat_layers.append(interp)
cat_layers = [input] + cat_layers cat_layers = [input] + cat_layers
out = fluid.layers.concat(cat_layers, axis=1, name='psp_cat') out = fluid.layers.concat(cat_layers, axis=1, name='psp_cat')
...@@ -344,7 +342,7 @@ class FastSCNN(object): ...@@ -344,7 +342,7 @@ class FastSCNN(object):
w = shape[-1] w = shape[-1]
h = shape[-2] h = shape[-2]
lower_res_feature = fluid.layers.resize_bilinear( lower_res_feature = fluid.layers.resize_bilinear(
lower_res_feature, [h, w], align_corners=False, align_mode=1) lower_res_feature, [h, w], align_mode=0)
with scope('dwconv'): with scope('dwconv'):
lower_res_feature = relu( lower_res_feature = relu(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册