2020-08-08

树叶识别

数据集如下图所示

不同类别之间的差别不是很明显,同一类别的树叶,形状不一
联系方式:1234585869(q号 添加麻烦备注 csdn)
在这里插入图片描述

在这里插入图片描述
使用matlab构建深度网络,
网络的输入即为图片的大小
网络最后的输出为类别数目
下面是网络的定义过程

%  构建网络模型           
layers = [imageInputLayer([250 250 3])     % 输入层
          convolution2dLayer(5,20)         % 卷积层
          reluLayer
          maxPooling2dLayer(2,'Stride',2)   % 池化层
          
          convolution2dLayer(5,20)    
          reluLayer
          maxPooling2dLayer(2,'Stride',2)
          
          convolution2dLayer(5,20)    
          reluLayer
          maxPooling2dLayer(2,'Stride',2)
          
          
          fullyConnectedLayer(184)        % 全链接层
          softmaxLayer
          classificationLayer()];          % 输出分类层
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18

此外也可以导入VGG16 VGG19 googlenet 等深度网络进行训练。
下图是我一些训练的过程

ALEXNET 冻结前面所有的参数-增大batch_size4——最终版在这里插入图片描述

ALEXNET 冻结前面10层的参数-增大batch_size3
在这里插入图片描述
VGG16训练的结果
在这里插入图片描述

VGG19 训练的结果
在这里插入图片描述

googlenet训练的结果googlenet训练的结果

©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页