我尝试创建一个图像分类器,但我不想使用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"?
1条答案
按热度按时间9njqaruj1#
根据the Tensorflow docs,tfds提供了现成的数据集,这意味着,没有任何条款可以上传你自己的数据集并使用它。
但是,这并不是道路的尽头,您可以遵循tensorflow中的tutorial。我将概述您必须用来创建数据集的重要代码。
这里根据您的训练和测试路径使用data_dir。
然后可以使用它来训练模型。