深度学习 - 将图像数据集分配到火车和测试中

Question

输入图像描述此处有3000张图像用于一个文件夹中的培训和测试，我在Label.csv文件中也有图像标签，该文件具有五个类别的类别。任何人都可以帮助我如何将该数据集分为火车和测试数据，以便我可以使用卷积神经网络对图像进行分类。与CSV和图像链接后，我的数据集看起来像以下图像。

Answer 1

首先，您需要图像和标签之间的关联（标签属于哪种图像的某种知识）。否则它将无法正常工作。之后，您可以将数据集拆分。这是一个玩具示例，假设full_dataset包含整个数据集，并且SIZE_OF_DATASET是full_dataset的大小：

full_dataset = full_dataset.shuffle()
train_dataset = full_dataset.take(int(0.8*SIZE_OF_DATASET))
test_dataset = full_dataset.skip(int(0.2*SIZE_OF_DATASET))