未验证 提交 1a30667d 编写于 作者: Y Yang Zhang 提交者: GitHub

Clean up pyreader vestiges (#228)

上级 2835d5ad
...@@ -216,7 +216,7 @@ ...@@ -216,7 +216,7 @@
"with fluid.program_guard(infer_prog, startup_prog):\n", "with fluid.program_guard(infer_prog, startup_prog):\n",
" with fluid.unique_name.guard():\n", " with fluid.unique_name.guard():\n",
" feed_vars = {\n", " feed_vars = {\n",
" var['name']: fluid.layers.data(\n", " var['name']: fluid.data(\n",
" name=var['name'],\n", " name=var['name'],\n",
" shape=var['shape'],\n", " shape=var['shape'],\n",
" dtype='float32',\n", " dtype='float32',\n",
...@@ -411,7 +411,7 @@ class CascadeMaskRCNN(object): ...@@ -411,7 +411,7 @@ class CascadeMaskRCNN(object):
'lod_level': 1 'lod_level': 1
} }
fields += box_fields fields += box_fields
feed_vars = OrderedDict([(key, fluid.layers.data( feed_vars = OrderedDict([(key, fluid.data(
name=key, name=key,
shape=inputs_def[key]['shape'], shape=inputs_def[key]['shape'],
dtype=inputs_def[key]['dtype'], dtype=inputs_def[key]['dtype'],
......
...@@ -312,7 +312,7 @@ class CascadeRCNN(object): ...@@ -312,7 +312,7 @@ class CascadeRCNN(object):
fields += ms_fields fields += ms_fields
self.im_info_names = ['image', 'im_info'] + ms_fields self.im_info_names = ['image', 'im_info'] + ms_fields
feed_vars = OrderedDict([(key, fluid.layers.data( feed_vars = OrderedDict([(key, fluid.data(
name=key, name=key,
shape=inputs_def[key]['shape'], shape=inputs_def[key]['shape'],
dtype=inputs_def[key]['dtype'], dtype=inputs_def[key]['dtype'],
......
...@@ -195,7 +195,7 @@ class CascadeRCNNClsAware(object): ...@@ -195,7 +195,7 @@ class CascadeRCNNClsAware(object):
use_dataloader=True, use_dataloader=True,
iterable=False): iterable=False):
inputs_def = self._inputs_def(image_shape) inputs_def = self._inputs_def(image_shape)
feed_vars = OrderedDict([(key, fluid.layers.data( feed_vars = OrderedDict([(key, fluid.data(
name=key, name=key,
shape=inputs_def[key]['shape'], shape=inputs_def[key]['shape'],
dtype=inputs_def[key]['dtype'], dtype=inputs_def[key]['dtype'],
......
...@@ -224,7 +224,7 @@ class FasterRCNN(object): ...@@ -224,7 +224,7 @@ class FasterRCNN(object):
fields += ms_fields fields += ms_fields
self.im_info_names = ['image', 'im_info'] + ms_fields self.im_info_names = ['image', 'im_info'] + ms_fields
feed_vars = OrderedDict([(key, fluid.layers.data( feed_vars = OrderedDict([(key, fluid.data(
name=key, name=key,
shape=inputs_def[key]['shape'], shape=inputs_def[key]['shape'],
dtype=inputs_def[key]['dtype'], dtype=inputs_def[key]['dtype'],
......
...@@ -314,7 +314,7 @@ class MaskRCNN(object): ...@@ -314,7 +314,7 @@ class MaskRCNN(object):
'lod_level': 1 'lod_level': 1
} }
fields += box_fields fields += box_fields
feed_vars = OrderedDict([(key, fluid.layers.data( feed_vars = OrderedDict([(key, fluid.data(
name=key, name=key,
shape=inputs_def[key]['shape'], shape=inputs_def[key]['shape'],
dtype=inputs_def[key]['dtype'], dtype=inputs_def[key]['dtype'],
......
...@@ -107,7 +107,7 @@ class RetinaNet(object): ...@@ -107,7 +107,7 @@ class RetinaNet(object):
use_dataloader=True, use_dataloader=True,
iterable=False): iterable=False):
inputs_def = self._inputs_def(image_shape) inputs_def = self._inputs_def(image_shape)
feed_vars = OrderedDict([(key, fluid.layers.data( feed_vars = OrderedDict([(key, fluid.data(
name=key, name=key,
shape=inputs_def[key]['shape'], shape=inputs_def[key]['shape'],
dtype=inputs_def[key]['dtype'], dtype=inputs_def[key]['dtype'],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册