未验证 提交 8684e869 编写于 作者: K Kaipeng Deng 提交者: GitHub

revert fit for resize_nearest scale. (#2242)

上级 15f4c254
...@@ -12,7 +12,7 @@ ...@@ -12,7 +12,7 @@
## Installation ## Installation
Running sample code in this directory requires PaddelPaddle Fluid v.1.5 and later. If the PaddlePaddle on your device is lower than this version, please follow the instructions in [installation document](http://www.paddlepaddle.org/documentation/docs/en/1.4/beginners_guide/install/index_en.html) and make an update. Running sample code in this directory requires PaddelPaddle Fluid v.1.4 and later. If the PaddlePaddle on your device is lower than this version, please follow the instructions in [installation document](http://www.paddlepaddle.org/documentation/docs/en/1.4/beginners_guide/install/index_en.html) and make an update.
## Introduction ## Introduction
......
...@@ -12,7 +12,7 @@ ...@@ -12,7 +12,7 @@
## 安装 ## 安装
在当前目录下运行样例代码需要PadddlePaddle Fluid的v.1.5或以上的版本。如果你的运行环境中的PaddlePaddle低于此版本,请根据[安装文档](http://paddlepaddle.org/documentation/docs/zh/1.4/beginners_guide/install/index_cn.html)中的说明来更新PaddlePaddle。 在当前目录下运行样例代码需要PadddlePaddle Fluid的v.1.4或以上的版本。如果你的运行环境中的PaddlePaddle低于此版本,请根据[安装文档](http://paddlepaddle.org/documentation/docs/zh/1.4/beginners_guide/install/index_cn.html)中的说明来更新PaddlePaddle。
## 简介 ## 简介
......
...@@ -46,9 +46,19 @@ def yolo_detection_block(input, channel, is_test=True, name=None): ...@@ -46,9 +46,19 @@ def yolo_detection_block(input, channel, is_test=True, name=None):
return route, tip return route, tip
def upsample(input, scale=2,name=None): def upsample(input, scale=2,name=None):
# get dynamic upsample output shape
shape_nchw = fluid.layers.shape(input)
shape_hw = fluid.layers.slice(shape_nchw, axes=[0], starts=[2], ends=[4])
shape_hw.stop_gradient = True
in_shape = fluid.layers.cast(shape_hw, dtype='int32')
out_shape = in_shape * scale
out_shape.stop_gradient = True
# reisze by actual_shape
out = fluid.layers.resize_nearest( out = fluid.layers.resize_nearest(
input=input, input=input,
scale=float(scale), scale=scale,
actual_shape=out_shape,
name=name) name=name)
return out return out
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册