如何通过 ImageDataGenerator 在 Kaggle/Colab 上使用 TPU？

Question

我有一个如下所示的目录和数据框：

|---classA
|-----1.jpg
|-----2.jpg
|---classB
|-----23.jpg
|-----124.jpg

数据框由两列组成：图像的名称及其 类

name  | class
1.jpg |   1
2.jpg |   1
23.jpg|   2

我用ImageDataGenerator，用的方法flow_from_directory。我目前正在使用 GPU 进行训练，但由于它很大，需要花费很多时间，所以我想切换到 TPU。我搜索了 Kaggle 和 Colab 的 TPU 指南，他们使用 TFRecord 来读取数据。谁能帮助我如何使用 ImageDataGenerator 中的 TFRecord？

Answer 1

所以我发现 ImageDataGenerator 不能在 TPU 上使用。您必须改用 tf.data.Dataset。