tf.keras.utils.image_dataset_from_directory 详细参数
时间: 2023-03-28 18:03:26 浏览: 206
图片dataset模型
tf.keras.utils.image_dataset_from_directory 是一个用于从目录中读取图像数据集的函数,其详细参数包括:
- directory:数据集所在目录的路径。
- labels:可选参数,指定数据集的标签。如果不指定,则默认使用目录名作为标签。
- label_mode:可选参数,指定标签的类型。可选值包括 "int"、"categorical"、"binary" 和 "sparse"。
- batch_size:每个批次的样本数。
- image_size:图像的大小,可以是一个整数或一个元组。
- color_mode:图像的颜色模式,可选值包括 "grayscale" 和 "rgb"。
- shuffle:是否在每个 epoch 开始时打乱数据集。
- seed:随机数种子。
- validation_split:可选参数,指定验证集的比例。
- subset:可选参数,指定使用数据集的子集。可选值包括 "training" 和 "validation"。
以上是 tf.keras.utils.image_dataset_from_directory 的详细参数。
阅读全文