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
1