上传者: 42134537
|
上传时间: 2022-03-23 15:21:16
|
文件大小: 13KB
|
文件类型: -
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