从字节对象使用MXRecordIO读取

时间:2019-02-28 21:48:32

标签: python-3.x mxnet bytesio

有没有一种方法可以使用mx.recordio.MXRecordIO来读取字节对象而不是文件对象?

例如,我目前正在做

import mxnet as mx

results_file = 'results.rec'
with open(results_file, 'wb') as f:
    f.write(results)

recordio = mx.recordio.MXRecordIO(results_file, 'r')
temp = recordio.read()

但是,如果可能的话,我宁愿不必写入文件作为中间步骤。我尝试使用BytesIO,但似乎无法使其正常工作。

1 个答案:

答案 0 :(得分:2)

目前,他们无法实现这一遗憾。这是不平凡的,因为RecordIO的读取/解析是用C ++完成的,您不能简单地将流转发到C ++ API。