tfrecord

[ API 文档 | crates.io ]

该软件包提供 TensorFlow TFRecord 数据格式序列化和反序列化的功能，并支持 TensorBoard！

• 支持 async/await 语法。易于与 futures-rs 一起使用。
• 与 serde、image、ndarray 和 tch 兼容。

许可

此软件以 MIT 许可证分发。请查看 LICENSE 文件以查看完整的许可证文本。

lib.rs:

该软件包提供将 TensorFlow 的 TFRecord 数据格式进行序列化和反序列化的功能。

Cargo 功能

可选功能

• full：启用所有功能。
• async：启用 async/await 功能。

第三方软件包支持

• with-serde：启用与 serde 的互操作性以序列化和反序列化示例类型。
• with-tch：启用 tch 类型支持。
• with-image：启用 image 类型支持。
• with-ndarray：启用 ndarray 类型支持。

手动生成 ProtocolBuffer 代码

该软件包编译TensorFlow中预生成的ProtocolBuffer代码。在TensorFlow版本更新时，您可能需要手动生成代码。它接受通过环境变量TFRECORD_BUILD_METHOD指定的几种访问TensorFlow源代码的方式。生成的代码放置在prebuild_src目录下。请参阅下面的示例，并将X.Y.Z替换为实际的TensorFlow版本。

从源tar包构建

export TFRECORD_BUILD_METHOD="src_file:///home/myname/tensorflow-X.Y.Z.tar.gz"
cargo build --release --features generate_protobuf_src

从源目录构建

export TFRECORD_BUILD_METHOD="src_dir:///home/myname/tensorflow-X.Y.Z"
cargo build --release --features generate_protobuf_src

从URL构建

export TFRECORD_BUILD_METHOD="url://https://github.com/tensorflow/tensorflow/archive/vX.Y.Z.tar.gz"
cargo build --release --features generate_protobuf_src

从系统上已安装的TensorFlow构建

构建脚本将在目录${install_prefix}/include/tensorflow中搜索protobuf文档。

export TFRECORD_BUILD_METHOD="install_prefix:///usr"
cargo build --release --features generate_protobuf_src