restore creator.py

gangliao-patch-1
gongweibao 8 years ago
parent 8c735c8b09
commit b359d5c5cd

@ -16,7 +16,7 @@ Creator package contains some simple reader creator, which could be used in user
program. program.
""" """
__all__ = ['np_array', 'text_file', "RecordIO"] __all__ = ['np_array', 'text_file']
def np_array(x): def np_array(x):
@ -55,22 +55,3 @@ def text_file(path):
f.close() f.close()
return reader return reader
def RecordIO(path):
"""
Creates a data reader that outputs record one one by one from given recordio file
:path: path of recordio file
:returns: data reader of recordio file
"""
def reader():
f = recordio.reader(path)
while True:
r = f.read()
if r is None:
break
yield r
f.close()
return reader

Loading…
Cancel
Save