未验证 提交 3acf9935 编写于 作者: C chengjuntao 提交者: GitHub

Fix fluid.data code example, test=develop, test=document_preview (#1472)

* fix fluid.data code example, test=develop, test=document_preview
上级 86067c9d
...@@ -45,56 +45,52 @@ deformable_roi_pooling ...@@ -45,56 +45,52 @@ deformable_roi_pooling
#position_sensitive为False #position_sensitive为False
import paddle.fluid as fluid import paddle.fluid as fluid
input = fluid.layers.data(name="input", input = fluid.data(name="input",
shape=[2, 192, 64, 64], shape=[2, 192, 64, 64],
dtype='float32', dtype='float32')
append_batch_size=False) rois = fluid.data(name="rois",
rois = fluid.layers.data(name="rois", shape=[-1, 4],
shape=[4], dtype='float32',
dtype='float32', lod_level=1)
lod_level=1) trans = fluid.data(name="trans",
trans = fluid.layers.data(name="trans", shape=[2, 384, 64, 64],
shape=[2, 384, 64, 64], dtype='float32')
dtype='float32', x = fluid.layers.deformable_roi_pooling(input=input,
append_batch_size=False) rois=rois,
x = fluid.layers.nn.deformable_roi_pooling(input=input, trans=trans,
rois=rois, no_trans=False,
trans=trans, spatial_scale=1.0,
no_trans=False, group_size=(1, 1),
spatial_scale=1.0, pooled_height=8,
group_size=(1, 1), pooled_width=8,
pooled_height=8, part_size=(8, 8),
pooled_width=8, sample_per_part=4,
part_size=(8, 8), trans_std=0.1,
sample_per_part=4, position_sensitive=False)
trans_std=0.1,
position_sensitive=False)
#position_sensitive为True #position_sensitive为True
import paddle.fluid as fluid import paddle.fluid as fluid
input = fluid.layers.data(name="input", input = fluid.data(name="input",
shape=[2, 192, 64, 64], shape=[2, 192, 64, 64],
dtype='float32', dtype='float32')
append_batch_size=False) rois = fluid.data(name="rois",
rois = fluid.layers.data(name="rois", shape=[-1, 4],
shape=[4], dtype='float32',
dtype='float32', lod_level=1)
lod_level=1) trans = fluid.data(name="trans",
trans = fluid.layers.data(name="trans", shape=[2, 384, 64, 64],
shape=[2, 384, 64, 64], dtype='float32')
dtype='float32', x = fluid.layers.deformable_roi_pooling(input=input,
append_batch_size=False) rois=rois,
x = fluid.layers.nn.deformable_roi_pooling(input=input, trans=trans,
rois=rois, no_trans=False,
trans=trans, spatial_scale=1.0,
no_trans=False, group_size=(1, 1),
spatial_scale=1.0, pooled_height=8,
group_size=(1, 1), pooled_width=8,
pooled_height=8, part_size=(8, 8),
pooled_width=8, sample_per_part=4,
part_size=(8, 8), trans_std=0.1,
sample_per_part=4, position_sensitive=True)
trans_std=0.1,
position_sensitive=True)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册