Блог

Я пытаюсь объединить два набора данных, которые были созданы с использованием keras.preprocessing.image_dataset_from_directory (), с путями следующим образом:

 >>> val_ds = tf.keras.preprocessing.image_dataset_from_directory(
    Val_data_dir,
    image_size=(127, 127),
    batch_size=32)
Found 16 files belonging to 2 classes.

>>> train_ds = tf.keras.preprocessing.image_dataset_from_directory(
    Train_data_dir,
    image_size=(127, 127),
    batch_size=32)
Found 5216 files belonging to 2 classes.

>>> print(len(list(train_ds)))
131
 

Когда я печатаю длину списка в train_ds, он показывает 131, когда я ожидаю 5216, которые он нашел при загрузке из каталога. Может кто-нибудь объяснить, почему? Какую еще функцию я должен использовать, чтобы получить количество файлов, загруженных в набор данных?

Чтобы получить количество всех файлов, вы можете использовать .file_paths атрибуты следующим образом:

 train_ds = tf.keras.preprocessing.image_dataset_from_directory(
    Train_data_dir,
    image_size=(32, 32),
    batch_size=32)

len(train_ds.file_paths)