Tensorflow 从图像制作训练测试数据集答案

【问题标题】：Tensorflow make Training Testing dataset from imagesTensorflow 从图像制作训练测试数据集
【发布时间】：2017-06-09 23:30:24
【问题描述】：

我正在制作一个图像分类器，我将 2 个类别的图像下载到 2 个文件夹：

├── Demo
│   ├── pizza
│   |── lasagnia

如何通过打乱图像并拆分为训练集和测试集，在 tensorflow 中制作图像训练和测试数据集。

【问题讨论】：

【解决方案1】：

以下应该有效： tf.split_v(tf.shuffle(images, ...), ...)

【讨论】：

【解决方案2】：

将工作目录设置为第一类的文件夹：

首先你需要下载库split-folders (pip install split-folders)

import os
import split_folders

cwd = os.getcwd()  

split_folders.ratio(cwd, output="output", seed=1337, ratio=(.8, .2))

对其他类文件夹重复。

【讨论】：