如何获取tf.keras.preprocessing.image_dataset_from_目录下的文件数量?
train_ds = tf.keras.preprocessing.image_dataset_from_directory(
data_dir,
validation_split=0.2,
subset="training",
seed=123,
image_size=(img_height, img_width),
batch_size=batch_size)
找到3670个文件,属于5个类。使用2936个文件进行训练。
3条答案
按热度按时间wxclj1h51#
当你跑的时候
这将为您提供文件数,在本例中为2936。
xj3cbfub2#
您可以使用 index_directory 帮助函数:
2vuwiymt3#
对于您的情况,这将工作:
在对象中查找元数据的一个技巧是设置一个断点并运行调试器。一旦对象被初始化,你就可以检查它是由什么组成的。
在这种情况下,你可以很容易地发现
train_ds
包含一个名为file_paths
的字段,进一步检查它,你会发现它只是一个普通的列表,我们可以很容易地从中提取长度。