【图像识别】VGG网络进行图像识别【附代码,数据集】
一.VGG 网络参数如下: VGG网络及使用的图像输入是3x224x224的图像。二.VGG 网络搭建如下(学习于B 站UP主:霹雳吧啦Wz,良心推荐):1.阅读代码之前了解下conv2d的计算,其实nn.Linear,nn.MaxPool2d的输出的计算都是使用以下公式:VGG16的输入是3x224x224,进入全连接层的输入是512x7x7,各个层次的输入输出,建议手算一遍。import torchimport torch.nn as nncfgs = { 'vgg11':
原创
2021-03-09 12:29:12 ·
5060 阅读 ·
1 评论