python-3.x - 分割图像数据集

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”中没有找到任何东西那么我在哪里存储拆分的输出（训练和测试）？

标签： python-3.xtensorflowkerassplit

Answer 1

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