Pytorch神经网络入门,使用MNIST数据集的手写数字识别,小白入门,尽量把每一步搞明白,记录遇到的问题。基于Pytorch简单实现了一个卷积神经网络CNN和前馈神经网络FNN。
MNIST数据集真的是是机器学习神经网络入门的经典数据集,这些工作最早都是深度学习三巨头之一Yann LeCun做的,在89年发表论文使用反向传播和神经网络识别手写数字,98年提出LeNet5卷积神经网络。数据集链接:http://yann.lecun.com/exdb/mnist/

一、神经网络

1.一般的神经网络

一般的卷积神经网络包括输入层、隐藏层和输出层,隐藏层可能会有多层,层与层之间是全连接的:在这里插入图片描述
输入和输出之间的关系为:
在这里插入图片描述
最终下一层的输出还要经过一个激活函数,直观的讲可以说是引入非线性:
在这里插入图片描述
在这里插入图片描述
本例中基于mnist数据集设计的前馈神经网络结构可以表示成这样:
在这里插入图片描述

2.卷积神经网络

相比于一般的神经网络,卷积神经网络是在隐藏层之前增加了卷积层和池化层。本例中结构:
在这里插入图片描述

3.反向传播bp算法

二、Pytorch实现cnn、fnn

Pytorch实现卷积神经网络mnist数据集手写识别

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐