基于Pytorch的LeNet实现CIFAR数据集分类(俺的代码注解)
开始了解深度学习有一小段事件了,期间我看了几个up主讲的LeNet实现,从最开始的每个函数都要百度到现在基本知道每部分的作用,还能比较熟练的搭建网络模型,我想记录一下一个菜鸟的成长经历。
代码是由b站up【霹雳吧啦Wz】的源码进行注解和稍微改动过的,初学者可以去看up的视频,讲的非常透彻
一、认识网络结构
首先我们开看一下LeNet的网络结构:
卷积+池化(下采样)+卷积+池化+三个全连接层
计算卷积输出大小的公式:N = (W − F + 2P )/S+1
N:图片输出的大小...
原创
2022-05-07 16:51:43 ·
389 阅读 ·
1 评论