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

问题描述

我们可以使用 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'

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）