- после перетасовки содержит 13 пакетов
- после пакетирования производит только 1 пакет (полагаю это оставшаяся партия)
- когда drop_remainder данные пусты
image_size = (300, 300)
batch_size = 32
train_dataset = image_dataset_from_directory(
dataset_dir,
image_size=(image_size[0], image_size[1]),
batch_size=batch_size,
label_mode="binary",
validation_split=0.2,
subset="training",
seed=123,
)
train_dataset = train_dataset.shuffle(1000)
train_dataset = train_dataset.batch(
batch_size=batch_size, drop_remainder=True
).prefetch(buffer_size=AUTOTUNE)
print(train_dataset.cardinality().numpy())
Подробнее здесь: https://stackoverflow.com/questions/793 ... -one-batch
Мобильная версия