为TPU加载图像数据的方法

Question

所以目前 keras.ImageDataGenerator 和 tf.data.Dataset 都不能使用 tensorflow TPU，有没有其他方法可以在 Tpu 上加载图像数据以训练模型？

Answer 1

您可以使用 GCS 存储桶。

或者您从文件中读取：

with open(image_path, "rb") as local_file:
  img = local_file.read()

然后转换为张量并使用 tf.data.Dataset.from_tensor_slices 创建数据集。