Numpy到TFRecords并返回

Question

如何将numpy数组转换为TFRecords？在运行模型时如何将其转换回来？我发现这个脚本https://gist.github.com/swyoon/8185b3dcf08ec728fb22b99016dd533f将numpy数组转换为TFRecords但我不明白如何从TFRecord中获取数据。

Answer 1

得到了!!! https://github.com/Hvass-Labs/TensorFlow-Tutorials/blob/master/18_TFRecords_Dataset_API.ipynb这个家伙Magnus Erik Hvass Pedersen有一个关于如何从TFRecords写入和读取的很棒的教程。他的脚本接收图像，但它可以通过替换numpy数组切片的图像轻松转换为np数组。