tensorflow实现步骤
1、准备数据
分别将待分类的图片数据分别单独放置到一个独立的文件夹中
例如:我需要进行五分类任务,我建立五个对应的文件夹,并且将每个类别的图片放置到自己的文件夹中,如下图所示

2、对数据进行预处理
由于所需分类图片大小不统一,需将图片同意裁剪为相同大小的图片进行,对应保存,格式和步骤一相同,如下图

3、生成pkl文件
将预处理的图片,进行pickle处理,将图片以及标签生成为pkl文件,如下图

4、搭建神经网络

本博客网络参考vgg论文:
点击打开链接
5、进行训练

6、测试训练model
代码:
代码链接