从 Amazon S3 路径读取 TFRecord 文件

Question

我尝试使用文件路径和 tf.data.TFRecordDataset() 直接从 Amazon S3 存储桶读取 TFRecord 文件。

似乎 tf.data.TFRecordDataset() 只接受 tf.string 或 tf.data.dataset 格式的文件名。

我尝试遵循，但没有成功。

filenames = ["s3://path_to_TFRecord"]
dataset = tf.data.TFRecordDataset(filenames)

我还尝试使用 s3fs 和文件句柄来解决该问题，但失败了。

任何解决方案表示赞赏。