网站做分站校园网络平台建设方案

卷积神经网络图像识别车辆类型

1、图像

自行车：

汽车：

摩托车：

2、数据集目录

3、流程

1、获取数据，把图像转成矩阵，并随机划分训练集、测试集
2、把标签转为数值，将标签向量转换为二值矩阵
3、图像数据归一化，0-1之间的值
4、构造卷积神经网络
5、设置图像输入形状(32, 32, 3)
6、设置卷积、池化层输出为三维矩阵
7、三维平展为一维,输入全连接层
8、输出层使用 softmax 激活函数，用于多分类问题
9、初始化 RMSprop 优化器
10、编译模型，多分类交叉熵损失函数，使用RMSprop自适应学习率的优化器，使用accuracy准确率作为衡量指标
11、添加训练轮数早停止机制
12、增加