自学内容网 自学内容网

每天五分钟深度学习pytorch:基于VGG神经网络完成CAFIR10的识别

本文重点

前面的所有模型我们都是使用VGG跑了mnist数据集,本次我们换一个数据集,我们使用CAFIR数据集,这个数据集我们前面介绍过,它和mnist不一样了,mnist是灰度图,这个是彩色图,所以它的通道数是3,这样我们再构建卷积神经网路的时候,第一个卷积层的输入通道数就应该是3而不是1了

VGG模型

跑模型(完成版)


原文地址:https://blog.csdn.net/huanfeng_AI/article/details/145330123

免责声明:本站文章内容转载自网络资源,如侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!