项目介绍

TensorFlow2.X 搭建卷积神经网络(CNN),实现水果识别。搭建的卷积神经网络是类似VGG的结构(卷积层与池化层反复堆叠,然后经过全连接层,最后用softmax映射为每个类别的概率,概率最大的即为识别结果)。

网络结构:

cnn 模型如何确定 cnn模型搭建_cnn 模型如何确定

开发环境:

  • python==3.7
  • tensorflow==2.3

数据集:

cnn 模型如何确定 cnn模型搭建_虚拟环境_02

cnn 模型如何确定 cnn模型搭建_虚拟环境_03


图片类别:

  • ‘freshapples’:‘新鲜苹果’,
  • ‘freshbanana’:‘新鲜香蕉’,
  • ‘freshoranges’:‘新鲜橙子’,
  • ‘rottenapples’:‘腐烂苹果’,
  • ‘rottenbanana’:‘腐烂香蕉’,
  • ‘rottenoranges’:‘腐烂橙子’

cnn 模型如何确定 cnn模型搭建_python_04

代码调试

cnn 模型如何确定 cnn模型搭建_tensorflow_05


拿到项目后,解压文件,解压后如下图所示:

cnn 模型如何确定 cnn模型搭建_python_06

Step1:打开项目文件夹

cnn 模型如何确定 cnn模型搭建_虚拟环境_07


各个文件及代码介绍:

cnn 模型如何确定 cnn模型搭建_cnn_08

Step2:搭建开发环境

cnn 模型如何确定 cnn模型搭建_python_09

创建虚拟环境

输入cmd回车后,会打开一个命令终端,下面我们开始创建虚拟环境:

cnn 模型如何确定 cnn模型搭建_虚拟环境_10


输入命令回车后,出现下面提示,继续回车:

cnn 模型如何确定 cnn模型搭建_tensorflow_11

输入命令为:

conda create -n FruitRec_py37 python=3.7

然后回车后,我们就创建了一个环境名称为“FruitRec_py37”的虚拟环境,它的python版本为3.7,如下图所示:

cnn 模型如何确定 cnn模型搭建_虚拟环境_12

激活虚拟环境

复制这条命令,输入命令行,激活我们创建的虚拟环境:

conda activate FruitRec_py37

cnn 模型如何确定 cnn模型搭建_cnn_13

安装第三方依赖库

下面开始安装项目用到的第三方依赖库,比如tensorflow、matplotlib、pyqt5等。本次用到的依赖库全部记录在了requirements.txt文件中。下面开始安装:

在命令终端中输入以下命令。

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

安装成功后,如下如所示:

cnn 模型如何确定 cnn模型搭建_cnn_14

打开项目配置环境

cnn 模型如何确定 cnn模型搭建_cnn_15

选择解释器(我们上面创建的虚拟环境)

cnn 模型如何确定 cnn模型搭建_cnn 模型如何确定_16


点击File,然后选择Settings

cnn 模型如何确定 cnn模型搭建_cnn_17


点击设置按钮,选择“Add”

cnn 模型如何确定 cnn模型搭建_cnn 模型如何确定_18


按照图片的提示操作即可,添加我们需要的“python解释器”,pycharm右下角显示如下图所示即为成功:

cnn 模型如何确定 cnn模型搭建_虚拟环境_19

训练神经网络模型

打开项目“train_cnn.py”这个代码,根据图片中提示操作:

cnn 模型如何确定 cnn模型搭建_tensorflow_20


成功运行效果如下图所示:

cnn 模型如何确定 cnn模型搭建_tensorflow_21


成功运行后,接下来要做的事情就是“等待”,根据每个人电脑配置的高低,运行训练网络的代码时间也是不同的(几分钟–几个小时)等待运行完毕,不报错即为训练成功。在训练成功后,在models文件夹中会生成"cnn_fv.h5"文件。

cnn 模型如何确定 cnn模型搭建_tensorflow_22


在训练成功后,在result文件夹中,可以看到“results_cnn.png”图片,记录了训练过程中准确率和loss的变化情况。

cnn 模型如何确定 cnn模型搭建_cnn_23

测试

在训练模型完毕后,我们开始测试模型(评估模型的性能),打开“test_model.py”

cnn 模型如何确定 cnn模型搭建_python_24


按照图片提示操作。运行成功后会在results文件夹下生成“heatmap_cnn.png”热力图(可以看到每个类别预测准确率的情况),如下所示:

cnn 模型如何确定 cnn模型搭建_cnn 模型如何确定_25

预测

在经过训练和测试之后,我们得到了一个可以用来做水果识别的神经网络,下面开始预测需要识别的水果图片。打开"windows.py"这个代码,直接点击运行,结果如下所示:

cnn 模型如何确定 cnn模型搭建_tensorflow_26

运行成功后得到一个pyqt5的GUI界面,然后我们就可以通过这个GUI操作去预测我们项目预测的水果图片啦!

cnn 模型如何确定 cnn模型搭建_cnn 模型如何确定_27