分割图像数据集

Question

我有一个图像数据集（jpg 图像文件夹）。我想拆分它：70% 用于训练，30% 用于随机测试。

所以我写了这个简单的脚本：

from sklearn.model_selection import train_test_split
path = ".\dataset"
output_split=train_test_split(path,path,test_size=0.2)

但我在“output_split”文件夹中找不到任何内容 那么我在哪里存储拆分的输出（训练和测试）？

Answer 1

我建议在您的项目中使用 tf.dataset。您可以在此处参考此链接以获取文档： https://www.tensorflow.org/api_docs/python/tf/keras/utils/image_dataset_from_directory 这个链接是一个函数使用的例子： https://keras.io/api/preprocessing/image/ 请记住对训练和（测试/验证）使用相同的种子。 请注意，您本地的文件不会发生任何更改。