分类：VGG11

2020
10-08

pytorch VGG11识别cifar10数据集(训练+预测单张输入图片操作)

首先这是VGG的结构图，VGG11则是红色框里的结构，共分五个block，如红框中的VGG11第一个block就是一个conv3-64卷积层：一，写VGG代码时，首先定义一个vgg_block(n,in,out)方法，用来构建VGG中每个block中的卷积核和池化层：n是这个block中卷积层的数目，in是输入的通道数，out是输出的通道数有了block以后，我们还需要一个方法把形成的block叠在一起，我们定义这个方法叫vgg_stack：defvgg_stack(num_convs,channels):#vgg_ne... 继续阅读 >