加载tpu图像数据的方法

问题描述

因此，目前keras.ImageDataGenerator和tf.data.Dataset都无法与tensorflow TPU一起使用，是否有其他方法可以在Tpu上加载用于训练模型的图像数据？

解决方法

您可以使用GCS存储桶。

或者您从文件中读取：

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

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