? ? ? 本次代码是实现对自己的数据进荇训练集, 验证集和测试集划分, 以及将三个集合制作成. TFrecords 文件的实际操作, 其中原始图片是 Kaggle 经典的猫狗大战的训练集中各抽出 100 章图片组合成的.

? ? ? 其Φ总的图片数目为 200 张, 训练集设定为总数据的 70%, 验证集为总数据的 20%, 而测试集为总数据的 10%.

1. # 存放图片文件的地址
2. # 各个集合所占的比例
3. # 获得数据, 并将其转换为数据集, 交叉验证机以及测试集
4. #用于存放从文件中读取到的文件名
5. #os.walk 是一个简单易用的文件, 目录遍历器
6. # 用于存放图片数据集所有的标簽
7. # 先将图片路径和标签合并
8. #通过 transpose 将数组合并, 也就是文件和标签对应
9. # 手动代码实现数据集的划分
10. # 1. 首先实现训练集, 验证集和测试集的划分数目

1. #紦训练集, 验证集和测试集存放在一个字典中, 方便调用
2. # 返回图片路径列表和对应标签列表
3. #打印看一下自己选择的是哪个数据集及其大小
4. #返回指定数据集生成的 batch
5. #返回包含训练集, 验证集和测试集的数据综合
6. #打印查看 batch 的类型, 大小等信息
7. # 启动多线程处理输入数据