基于Alexnet的植物病斑识别

最新推荐文章于 2023-02-01 15:17:59 发布

Make Huang

最新推荐文章于 2023-02-01 15:17:59 发布

阅读量2.6k

点赞数 1

分类专栏：图像分类

本文链接：https://blog.csdn.net/hxf19961219/article/details/89298233

版权

a.准备数据
数据使用的是一个植物病害检测比赛中的数据，选择了四类苹果叶片进行识别，真实数据如下所示：
在这里插入图片描述
b.模型结构
本次实验我使用的是Alexnet网络，网络在2012年被提出。网络结构如下：

根据Alexnet网络结构，我使用Keras打印了模型信息如下：

c.代码实现
使用keras实现了Alexnet完整代码，数据放在train和test下，每个文件夹下有四个子文件夹，病害类别标签使用0、1、2、3统一标签。

from keras.preprocessing.image import ImageDataGenerator
from keras.models import Sequential
from keras.layers import Conv2D, MaxPooling2D
from keras.layers import Activation, Dropout, Flatten, Dense
from keras import backend as K


# dimensions of our images.
img_width, img_h