代码使用Pytorch构建GoogLeNet网络,Pytorch可从官网查看下载方式https://pytorch.org/get-started/locally/ torchinfo可由pip install下载,主要用来显示网络具体架构。
'GoogLeNet.pdf'是原论文。
'GoogLeNet.png'展示了GoogLeNet的具体架构,主要亮点就是加入了Inception层。
'GoogLeNet参数.png'展示了每一层结构的具体参数,代码中会用到。
'GoogLeNet.py'中包含了四个类:'BasicConv2d'是基础的卷积层,'Inception'就是Inception层,'Auxclf'是辅助分类器,'GoogLeNet'就是GoogLeNet网络主体。