Matlab使用CNN卷积神经网络进行图像分类,使用了猫狗大战数据集的4000个图像(2000猫2000狗),分为猫狗两个类别。
也可以改成多分类。
注释详细,可直接运行,可以直接换成自己的数据,源代码和数据文件都会发给你。
工作如下:
1、加载数据集,并划分,70%训练,10%验证,20%测试。
数据量一共为4000(2000猫2000狗)。
2、搭建CNN网络,网络为三层,构建优化器。
3、训练
4、测试,计算准确率,绘制混淆矩阵。
注:本程序只用于教学练习如何在Matlab里使用CNN,由于数据量太少和网络简单,准确率结果并不是很好。
4000样本时,GPU运行时间大约7分钟。
为了方便没有GPU的同学学习,又设计了CPU版本。
不用GPU的版本:只用了1000个样本(500猫500狗),在CPU上训练大约用时15分钟。
网络层数也降低为了两层,且没有使用验证集。
ID:4715
详询客服 微信shujuqudong1 或shujuqudong6 或 qq68823886 或 27699885
图文详情请查看: http://matup.cn/675932521785.html