Python 如何获取TFRecord文件中包含的条目总数？

我能够使用tensorflow编写和读取TFrecord文件。如何快速获取TFRecord文件中包含的条目总数？是否有任何API来获取计数？

基本上是一系列结构，其形式如下：

struct TFRecords {
    uint64_t length;
    uint32_t length_checksum;
    uint8_t data[length];
    uint32_t data_checksum;
} × N

没有元数据来告诉有多少条记录，因此获取总数的唯一可靠方法是读取整个文件（因此没有API来快速获取总数），然后调用

num\u records\u producted（）

您可以编写包含该编号的自定义元数据，作为生成TFR记录时的第一条记录

---

如果您确定每个记录的长度都相同，那么您可以得到条目数，即

解压缩的\u文件大小/（每个\u记录的长度+16）

谢谢，我真的应该在文件名中包含计数。