TensorFlow - 如何读取可变长度的数据部分?

时间:2017-01-25 08:44:50

标签: python tensorflow

我想在TensorFlow中使用数据读取器读取单个文件的variable_length部分数据。 我的文件如下:[data_length_1,data_bytes_1,data_length_2,data_bytes_2,...] 第一个字节表示要读取的字节数,然后重复此过程。

问题是我想在TensorFlow中将数据读入队列。我尝试了tf.FixedLengthRecordReader,但它只接受要读取的固定长度数据。有没有办法将其改成可变长度数据读取器?

1 个答案:

答案 0 :(得分:0)

我设法通过添加custom data reader来解决我的问题,该{{3}}从第一个数据字节读取每条记录的记录长度,并且可以正常工作。