如何在 Pandas 数据框中导入 tfrecord 文件？

Question

我有一个 tfrecord 文件，想将它导入到 Pandas 数据框或 numpy 数组中。

我找到了读取 tfrecords 的工具，但它们只能在 tensorflow 会话中工作，这不是我的用例......

感谢我能得到的任何帮助！

Answer 1

在 Colab 中你可以输入（或者在你的 cmd 上不输入！）

!pip install pandas-tfrecords

安装后您可以使用：

import pandas as pd
import pandas_tfrecords as pdtfr
pdtfr.tfrecords_to_pandas(file_paths=r'/folder/file.tfrecords')

祝你好运！