train_ds = tf.keras.preprocessing.image_dataset_from_directory
时间: 2023-08-11 09:03:05 浏览: 142
tf.keras_分类模块数据归一化.ipynb_quietlyxoc_keras_归一化_lateory_tf.keras_源
该函数是 TensorFlow 中用于从文件夹中加载图像数据集的方便函数。它返回一个 tf.data.Dataset 对象,该对象包含从文件夹中加载的所有图像及其相应的标签。可以通过指定文件夹路径、目标图像大小、颜色模式、批处理大小等参数来自定义该函数的行为。例如:
```
train_ds = tf.keras.preprocessing.image_dataset_from_directory(
"path/to/folder",
image_size=(224, 224),
batch_size=32,
validation_split=0.2,
subset="training",
seed=123
)
```
这将从名为 "path/to/folder" 的文件夹中加载图像数据集,将其大小调整为 (224, 224),每个批次包含 32 张图像,其中 80% 用于训练,20% 用于验证。最后,可以使用 `train_ds` 对象来训练模型。
阅读全文