分类：CIFAR10

2020
10-08

keras实现VGG16 CIFAR10数据集方式

我就废话不多说了，大家还是直接看代码吧！importkerasfromkeras.datasetsimportcifar10fromkeras.preprocessing.imageimportImageDataGeneratorfromkeras.modelsimportSequentialfromkeras.layersimportDense,Dropout,Activation,Flattenfromkeras.layersimportConv2D,MaxPooling2D,BatchNormalizationfromkerasimportoptimizersimportnumpyasnpfromkeras.layers.coreimportLambdafromkerasimpor... 继续阅读 >

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... 继续阅读 >