keras 在google中使用自己的数据集和tfds. load

pbossiut  于 2022-11-24  发布在  Go
关注(0)|答案(1)|浏览(170)

我尝试创建一个图像分类器,但我不想使用TensorFlow已经拥有的数据集(在本例中为"bean"),而是想使用我自己的数据集。
作为参考,这被称为"flower-detection",图像被分为"training"和"testing"。我已经把它上传到Google云端硬盘并安装了它,但是我不知道如何在tfds.load中使用它。我正在学习一个教程,所以有没有办法把下面这行代码改为使用我自己的数据集?

ds_test = tfds.load(name="beans", split="test")

如何将数据集从Google云端硬盘导入到tfds.load中?我是将名称写为"flower-detection"的完整文件夹放入tfds.load中,还是分别放入"flower-detection/training"和"flower-detection/testing"?

9njqaruj

9njqaruj1#

根据the Tensorflow docs,tfds提供了现成的数据集,这意味着,没有任何条款可以上传你自己的数据集并使用它。
但是,这并不是道路的尽头,您可以遵循tensorflow中的tutorial。我将概述您必须用来创建数据集的重要代码。

train_ds = tf.keras.preprocessing.image_dataset_from_directory(
  data_dir,
  subset="training",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

val_ds = tf.keras.preprocessing.image_dataset_from_directory(
  data_dir,
  subset="validation",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

这里根据您的训练和测试路径使用data_dir。
然后可以使用它来训练模型。

history = model.fit(
  train_ds,
  validation_data=val_ds,
  epochs=epochs
)

相关问题