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

【问题标题】：How to use TPU on Kaggle/Colab with ImageDataGenerator?如何通过 ImageDataGenerator 在 Kaggle/Colab 上使用 TPU？
【发布时间】：2020-09-29 14:22:34
【问题描述】：

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

|---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？

【问题讨论】：

AFAIK，你不能在 TPU 上使用 ImageGens！
是的，我刚刚找到了 Kaggle 工作人员关于 TPU 上 IDG 的评论。无论如何我可以以某种方式“复制” IDG？

标签： tensorflow keras tpu

【解决方案1】：

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

【讨论】：