如何使用 tf.keras.preprocessing.image_dataset_from_directory 获取类的数量？

Question

img_height,img_width=180,100 batch_size=32 train_ds = tf.keras.preprocessing.image_dataset_from_directory(data_dir1,validation_split=0.01,subset="training",seed=123,image_size=(img_height, img_width),batch_size=batch_size)

输出：找到属于 4 个类的 1376 个文件。 使用 1363 文件进行训练。

如何获取变量中的类总数？

Answer 1

如果你有类似的东西

train_gen=tf.keras.preprocessing.image_dataset_from_directory(etc

那么您可以使用下面的代码来获取您想要的信息类型

classes=list(train_gen.class_indices.keys())
class_indices=list(train_gen.class_indices.values())
num_of_classes=len(classes)

train_gen.class_indices 是 {class: index} 形式的字典

Answer 2

label_map = (train.ds.class_indices)