gpt4 book ai didi

tensorflow - 是否有用于解析 TFRecord 文件的 TensorFlow C++ API?

转载 作者:行者123 更新时间:2023-12-03 18:02:07 26 4
gpt4 key购买 nike

有什么等同于 tf.python_io.tf_record_iterator() 的东西吗?和 ParseFromString() 功能?谷歌搜索并查看 TensorFlow 存储库并没有出现太多结果。

最佳答案

TFRecord 的每个条目都按以下方式组织(您可以通过 tensorflow 存储库中的 record_writer.h 进行验证)

class RecordWriter {
public:
// Format of a single record:
// uint64 length
// uint32 masked crc of length
// byte data[length]
// uint32 masked crc of data
static constexpr size_t kHeaderSize = sizeof(uint64) + sizeof(uint32);
static constexpr size_t kFooterSize = sizeof(uint32);
整个 TFRecord 只是上述结构的重复。
  • 两人份 crc字段,你可以跳过它们(小心)
  • 对于 length字段,您可以在 C++ 中将 8 个字节读入 uint64_t 变量
  • 对于 data字段,其实就是序列化的Protocal Buffer message
  • message的格式在 example.proto 中描述在tensorflow存储库中,您可以使用相应版本的protoc对其进行编译并获取API来解码序列化 data .
    在我的 github中给出了更详细的解释和相应的代码,你可以试一试:)

    关于tensorflow - 是否有用于解析 TFRecord 文件的 TensorFlow C++ API?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49558269/

    26 4 0
    Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
    广告合作:1813099741@qq.com 6ren.com