Matlab使用CNN卷积神经网络进行图像分类,使用了猫狗大战数据集的4000个图像(2000猫2000狗),分为猫狗两个

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

QQ客服:27699885 微信客服:shujuqudong1 微信客服:shujuqudong6


本站资料较杂,精品资料单独做了汇总,可查阅 https://liruan.net/ 里面资源更优秀