python - 如何知道 tf.data.Dataset.from_generator() 中的图像数量

Question

我们可以使用 ImageDataGenerator 和 tf.data.Dataset.from_generator() 方法生成图像数据集。

img_gen = tf.keras.preprocessing.image.ImageDataGenerator()
gen = img_gen.flow_from_directory(path)
train_set = tf.data.Dataset.from_generator(gen, ...)
train_set.shuffle(2)

现在，我想知道 train_set 中的图像数量，我这样做：

train_set.samples

错误是：

AttributeError: 'ShuffleDataset' object has no attribute 'samples'

标签： pythonkerasdata-generation