这份数据集来源于Kaggle,数据集有12500只猫和12500只狗。在这里简单介绍下整体思路
处理数据
设计神经网络
进行训练测试
1. 数据处理
将图片数据处理为 tf 能够识别的数据格式,并将数据设计批次。
第一步get_files() 方法读取图片,然后根据图片名,添加猫狗 label,然后再将 image和label 放到 数组中,打乱顺序返回
将第一步处理好的图片 和label 数组 转化为 tensorflow 能够识别的格式,然后将图片裁剪和补充进行标准化处理,分批次返回。
新建数据处理文件 ,文件名 input_data.py
import tensor
2021-03-22 18:09:17
80KB
ns
OR
ow
1