tfrecord:PyTorch的TFRecord阅读器

时间:2024-02-27 01:14:28
【文件属性】:

文件名称:tfrecord:PyTorch的TFRecord阅读器

文件大小:13KB

文件格式:ZIP

更新时间:2024-02-27 01:14:28

tensorflow loader pytorch dataset tfrecord

TFRecord阅读器 安装 pip3安装tfrecord 用法 建议为每个TFRecord文件创建一个索引文件。 使用多个工作程序时必须提供索引文件,否则加载程序可能会返回重复的记录。 python3 -m tfrecord.tools.tfrecord2idx 使用TFRecordDataset读取PyTorch中的TFRecord文件。 import torch from tfrecord . torch . dataset import TFRecordDataset tfrecord_path = "/path/to/da


【文件预览】:
tfrecord-master
----tfrecord()
--------iterator_utils.py(2KB)
--------example_pb2.py(17KB)
--------tools()
--------__init__.py(279B)
--------writer.py(3KB)
--------reader.py(8KB)
--------torch()
----MANIFEST(273B)
----LICENSE(1KB)
----setup.py(570B)
----.gitignore(29B)
----README.md(3KB)

网友评论