最简单的利用tensorflow实现卷积神经网络分类任务(ubuntu windows都适用)提供代码

tensorflow实现步骤

1、准备数据

分别将待分类的图片数据分别单独放置到一个独立的文件夹中

例如:我需要进行五分类任务,我建立五个对应的文件夹,并且将每个类别的图片放置到自己的文件夹中,如下图所示

2、对数据进行预处理

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


3、生成pkl文件

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


4、搭建神经网络

本博客网络参考vgg论文:

点击打开链接

5、进行训练


6、测试训练model


代码:

代码链接

猜你喜欢

转载自blog.csdn.net/qq_34654240/article/details/80047887