利用预训练的VGG提取图像特征(pytorch)
核心部分a=vgg.features(img)#使用feature获取特征层(卷积层)的特征;输出特征维度为【1,512,4,4】b=vgg.avgpool(a)#使用vgg定义的池化操作;输出特征维度为【1,512,7,7】b=torch.flatten(b,1)#将特征变成一维度;输出特征维度为【1,25088】c=vgg.classifier[:1]#使用分类层的的第一层,当然可以选择数;输出特征维度为【1,4096】前言需要pytorch预训练模型VGG19提取图像特征,但VGG1
原创
2022-02-27 13:11:20 ·
6804 阅读 ·
0 评论