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

问题描述

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

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

解决方法

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