当前位置：首页 > news >正文

VGGNet

news 文章来源：https://blog.csdn.net/2303_79620604/article/details/139398420 2025/5/4 9:25:43

VGGNet

CNN卷积网络的发展史

1. LetNet5(1998)
2. AlexNet(2012)
3. ZFNet(2013)
4. VGGNet(2014)
5. GoogLeNet(2014)
6. ResNet(2015)
7. DenseNet(2017)
8. EfficientNet(2019)
9. Vision Transformers(2020)
10. 自适应卷积网络(2021)

上面列出了发展到现在CNN的一些经典的网络模型，我将持续不断更新学习上述神经网络的笔记。共勉！

原论文地址VGGNet

关键知识点

使用3 * 3的卷积核，增加网络深度用于提取更多特征，提高分类准确度。
使用尺度抖动增强训练集，提高模型准确率
密集评估，多裁剪评估，单尺度评估，多尺度评估，多模型融合

1. 简介

在这里插入图片描述

本文研究了在大型图像识别设置中，卷积网络深度对准确性的影响。我们的主要贡献是通过对具有非常小的（3×3）卷积滤波器的架构进行彻底的深度评估，深度增加到16-19个权重层，这表明通过将深度推向16-19个权重层，可以获得对先前的配置的显着改进。这些发现是我们在2014年ImageNet挑战赛中的主要贡献，我们的团队在本地化和分类轨道上分别获得了第一名和第二名。我们还展示了我们的表示在其它数据集上具有良好的泛化性，在那里它们获得了最先进的结果。我们已将我们的两个最佳表现卷积网络模型公开可用，以促进进一步研究深度视觉表示在计算机

2. ConvNet Configurations(卷积配置)

2.1 Architecture（结构）

在这里插入图片描述

输入的是224224的RGB图片，经过堆叠的3 * 3卷积核（为什么这个小的卷积核是33不是22或者是11呢，因为只有3*3是捕捉画面中心左右上下的最小尺寸）步长是1(也叫特征精度),对卷积层的输入进行填充为了保持卷积后的空间分辨率保持不变，padding=1，使用了5个池化层，最大池化由2 * 2像素，步长为2。3个全连接层，前两个都有4096个神经元，最后一个有1000个神经元。所有隐层都使用Relu激活函数，除了一个其他的所有网络都没有包含局部响应归一化在2.4节说明了，这种并不能显著提升性能还会导致内存消耗计算时间增加。

在这里插入图片描述

在上一节的ZFNet网络中我们知道，过大卷积核提取到的feature存在高频或低频这种无效信息，所以VGGNet网络中使用3*3的小卷积核，这样能够提取到更多的有效信息。

2.2 Configurations（网络配置）

在这里插入图片描述

表1：从A-E不停的增加网络深度（11层到16再到19层），每次经过最大池化后卷积的通道数成倍增加直到512为止。

在这里插入图片描述

在表2中，我们报告了每个配置的参数数量。尽管深度很大，但我们的网络中的权重数量并不大于具有更大卷积层宽度和感受野的更浅的网络中的权重数量((Sermanet et al.， 2014)中的144M权重)。

2.3 Discussion(讨论)

在这里插入图片描述

采用3*3的卷积核：

3*3层卷积核可以捕捉3中不同的feature
可以减少参数数量

例如：采用3层3 * 3的卷积核参数数量为 $3（3^2C^2）= 27C^2$ ,采用7 * 7为： $7^2C^2 = 49C^2$
这里表1中配置C采用添加了一层1 * 1的卷积核是为了增加决策函数非线性而不影响函数感受野（就是Relu函数）这里重点用的是D的配置。

3. Classification Framework（分类框架）

3.1 Training（训练细节）

使用的是有动量的小批次梯度下降，batch size是256，动量是0.9，权重是 $5*10^{-4}$ ,dropout为0.5，学习率learning rate为 $10^{-2}$ ,训练过程中学习率一共下降了3次，经过37w次迭代74轮。
先随机初始化浅层的网络然后逐渐将网络的层数增加，增加的部分参数随机初始化其他部分继承前面网络。进行了水平翻转和颜色变换进行数据增强。
模型分为单尺度和多尺度结构（模型可以接受多个size的图像），单尺度模型训练时可以先训练小尺度然后再训练大尺度来增加训练速度。

对于选择softmax分类器还是k个logistics分类器，取决于所有类别之间是否互斥。所有类别之间明显互斥用softmax；所有类别之间不互斥有交叉的情况下最好用k个logistics分类器

训练图片预处理过程：
1.训练图片归一化，图像等轴重调(最短边为S)
等轴重调剪裁时的两种解决办法：
(1) 固定最小遍的尺寸为256
(2) 随机从[256,512]的确定范围内进行抽样，这样原始图片尺寸不一，有利于训练，这个方法叫做尺度抖动，有利于训练集增强。训练时运用大量的裁剪图片有利于提升识别精确率。

2.随机剪裁(每SGD一次)
3.随机水平翻转
4.RGB颜色偏移

3.2 Testing(测试细节)

密集评估(Dense Evaluation)： 在VGGNet的测试阶段，首先将网络转化为全卷积网络。第一个全连接层转为7×7的卷积层，后两个全连接层转化为1×1的卷积层。这种转换能使得网络能够接受任意尺寸的输入图像，并在整个图像上个密集的应用卷积操作。最终，网络输出一个类被分数图，其尺寸取决于输入图像的调整尺寸Q。每个类别的得分是通过对这个分数图进行空间平均得到的。
多裁剪评估(Multi-crop Evaluation)： VGGNet还使用了多裁剪评估方法，即在测试时对输入图像进行多次不同区域的裁剪，并对每个裁剪的图像分别应用于网络。这种方法通常包括对图像的四个角落和中心区域进行裁剪，有时还包括它们的水平翻转版本。每个裁剪得到的预测结果最后被合并，以提高分类的准确性。
测试集增强： 我们还通过水平翻转图像来增强测试集，运用原始图片的softmax的后验概率以及其水平翻转的平均来获得图片的得分

3.3 实验

单尺度评估

在固定的训练和测试图像尺寸S=Q和变化的训练图像尺寸 $S\in[S_{min},S_{max}],Q = 0.5(S_{min}+S_{max})$ ,在不同深度的模型A-E上的测试结果，如下所示：

在这里插入图片描述

从结果我们可以发现通过尺寸变化的训练集增强有助于捕获多尺度的图像统计。

多尺度评估

评估测试时尺寸变化的影响，对于固定的训练尺寸S，测试时采用 $Q\in \{S-32,S+32\}$ ,对于变化的训练尺寸 $S\in[S_{min},S_{max}]$ ,测试尺寸采用 $Q = \{S_{min},0.5(S_{min}+S_{max}),S_{max}\}$ 。在模型A-E上的测试结果如下所示：

在这里插入图片描述

测试时的尺寸变化能够取得更好的性能

多裁剪评估

1.上面我们放弃了测试时剪裁图像，但是我们也觉得剪裁图片有一定的效果，因为同时，如Szegedy等人（2014）所做的那样，使用大量的裁剪图像可以提高准确度。
他们成功的原因是：
1.因为与全卷积网络相比，它使输入图像的采样更精细。
2.此外，由于不同的卷积边界条件，多裁剪图像评估是密集评估的补充：当将ConvNet应用于裁剪图像时，卷积特征图用零填充，而在密集评估的情况下，相同裁剪图像的填充自然会来自于图像的相邻部分（由于卷积和空间池化），这大大增加了整个网络的感受野，因此捕获了更多的上下文。

2.虽然我们认为在实践中，多裁剪图像的计算时间增加并不足以证明准确性的潜在收益（计算消耗的时间带来了准确性的提高不足以说明它对网络来说是一种增益），但作为参考，我们还在每个尺度使用50个裁剪图像（5×5规则网格，2次翻转）评估了我们的网络，在3个尺度上总共150个裁剪图像。这种测试方法，和训练过程类似，不用将网络转化为全卷积网络，是从Q×Q大小的图中随机裁剪224×224的图作为输入，文中裁剪了50个图像，而每个图像之前缩放为三个尺寸，所以每个图像的测试图像数目变为150个，最终结果就是在150个结果中取平均。

3.再加上原来的密集评估和两者的结合（即两者结果取平均，其能大大增加网络的感受野，因此捕获更多的上下文信息，实验中也发现这种方法表现最好。）
测试结果如下：
在这里插入图片描述

这一结果说明了两种评估方法有一定的互补性。（利用模型的互补性提高模型的性能，这也是现在比赛中参赛者所做的

多模型融合

在这里插入图片描述

通过对多个不同ConvNet模型的softmax类后验概率进行平均，来组合这些模型的输出。由于不同模型之间的互补性，这种做法可以改进性能。作者将表现最佳的两个模型（D和E）进行组合，得到了top-5错误率6.8%的效果。

与最新模型进行比较

在这里插入图片描述

从结果可知显著优于上一代模型，并在单网络性能上取得了最优的结果。

5. Conclusion(总结)

采用多个小的卷积核但是层数增加卷积核越多可以提取到更多的特征越多，所以网络的深度对识别结果，检测结果都非常有效，可以捕获更多的信息。

Pytorch实现

实现步骤：

导入所需要的库
导入pytorch中自带的vgg19网络模型参数
读取图片
对图片进行预处理
将图片喂入网络中并进行可视化
预测结果

import numpy as np
import torch
import torchvision
from torchvision import models,transforms
from PIL import Image
import torch.nn.functional as F
import pandas as pd
import matplotlib.pyplot as plt
import requests
import cv2
import json

use_pretrained = True
net = models.vgg19(pretrained=use_pretrained)
net.eval()print(net)

c:\Users\10766\software\Anaconda\envs\pytorch\lib\site-packages\torchvision\models\_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead.warnings.warn(
c:\Users\10766\software\Anaconda\envs\pytorch\lib\site-packages\torchvision\models\_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=VGG19_Weights.IMAGENET1K_V1`. You can also use `weights=VGG19_Weights.DEFAULT` to get the most up-to-date weights.warnings.warn(msg)VGG((features): Sequential((0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(1): ReLU(inplace=True)(2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(3): ReLU(inplace=True)(4): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)(5): Conv2d(64, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(6): ReLU(inplace=True)(7): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(8): ReLU(inplace=True)(9): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)(10): Conv2d(128, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(11): ReLU(inplace=True)(12): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(13): ReLU(inplace=True)(14): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(15): ReLU(inplace=True)(16): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(17): ReLU(inplace=True)(18): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)(19): Conv2d(256, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(20): ReLU(inplace=True)(21): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(22): ReLU(inplace=True)(23): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(24): ReLU(inplace=True)(25): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(26): ReLU(inplace=True)(27): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)(28): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(29): ReLU(inplace=True)(30): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(31): ReLU(inplace=True)(32): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(33): ReLU(inplace=True)(34): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))(35): ReLU(inplace=True)(36): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False))(avgpool): AdaptiveAvgPool2d(output_size=(7, 7))(classifier): Sequential((0): Linear(in_features=25088, out_features=4096, bias=True)(1): ReLU(inplace=True)(2): Dropout(p=0.5, inplace=False)(3): Linear(in_features=4096, out_features=4096, bias=True)(4): ReLU(inplace=True)(5): Dropout(p=0.5, inplace=False)(6): Linear(in_features=4096, out_features=1000, bias=True))
)

img = Image.open('./pre_images/Golden_Retriever.jpg')
# 将img转换为numpy数组，并除以255，使其值范围在0到1之间
imarray = np.asarray(img) / 255.0
plt.figure()
plt.imshow(imarray)
plt.axis('off')
plt.show()

在这里插入图片描述

data_transform = transforms.Compose([transforms.Resize((224,224)),transforms.ToTensor(),transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])input_img = data_transform(img).unsqueeze(0)
print(input_img.shape)

torch.Size([1, 3, 224, 224])

activation = {}
def get_activation(name):def hook(model, input, output):activation[name] = output.detach()return hook

net.features[4].register_forward_hook(get_activation('maxpool1'))
_ = net(input_img)
maxpool1 = activation['maxpool1']
print(maxpool1.shape)

torch.Size([1, 64, 112, 112])

plt.figure(figsize=(11,6))
for i in range(maxpool1.shape[1]):plt.subplot(6,11,i+1)plt.imshow(maxpool1.data.numpy()[0,i,:,:],cmap='gray')plt.axis('off')plt.subplots_adjust(wspace=0.1, hspace=0.1)
plt.show()

在这里插入图片描述

net.eval()
net.features[21].register_forward_hook(get_activation('conv21'))
_ = net(input_img)
conv21 = activation['conv21']
print(conv21.shape)

torch.Size([1, 512, 28, 28])

plt.figure(figsize=(11,8))
for i in range(88):plt.subplot(8,11,i+1)plt.imshow(conv21.data.numpy()[0,i,:,:],cmap='gray')plt.axis('off')plt.subplots_adjust(wspace=0.1, hspace=0.1) 
plt.show()

在这里插入图片描述

ImageNet标签下载地址

jsonfile = r"./data/ImageNet-1K-main/ImageNet_class_index.json"
with open(jsonfile,'r') as load_f:load_json = json.load(load_f)labels = {int(key):value for (key,value) in load_json.items()}

net.eval()
img_pre = net(input_img)
softmax = torch.nn.Softmax(dim=1)
im_pre_prob = softmax(img_pre)
prob,prelab = torch.topk(im_pre_prob,5)
prob = prob.data.numpy().flatten()
prelab = prelab.numpy().flatten()
for i,lab in enumerate(prelab):print("index: {}, label: {}, probability: {}".format(lab,labels[lab][1],prob[i]))

index: 207, label: golden_retriever, probability: 0.873153567314148
index: 852, label: tennis_ball, probability: 0.02229723334312439
index: 208, label: Labrador_retriever, probability: 0.016353365033864975
index: 220, label: Sussex_spaniel, probability: 0.009220004081726074
index: 168, label: redbone, probability: 0.006000933237373829

通过结果可知预测的结果为golden_retriever，预测结果很正确。

接下里我们再预测一张图片，看看结果是否正确

img_1 = Image.open('./pre_images/1.jpg')
imarray = np.asarray(img_1) / 255.0
plt.figure()
plt.imshow(imarray)
plt.axis('off')
plt.show()

在这里插入图片描述

data_transform = transforms.Compose([transforms.Resize((224,224)),transforms.ToTensor(),transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])input_img_1 = data_transform(img_1).unsqueeze(0)
print(input_img_1.shape)

torch.Size([1, 3, 224, 224])

net.eval()
net.features[21].register_forward_hook(get_activation('conv21'))
_ = net(input_img_1)
conv21 = activation['conv21']
print(conv21.shape)

torch.Size([1, 512, 28, 28])

plt.figure(figsize=(11,8))
for i in range(88):plt.subplot(8,11,i+1)plt.imshow(conv21.data.numpy()[0,i,:,:],cmap='gray')plt.axis('off')plt.subplots_adjust(wspace=0.1, hspace=0.1)
plt.show()

在这里插入图片描述

img_pre_1 = net(input_img_1)  # 预测输入图片的类别
softmax = torch.nn.Softmax(dim=1)  # 定义softmax函数
im_pre_prob_1 = softmax(img_pre_1)  # 对预测结果进行softmax激活
prob_1,prelab_1 = torch.topk(im_pre_prob_1,5)  # 获取概率最大的5个类别及其概率
prob_1 = prob_1.data.numpy().flatten()  # 将概率值转换为numpy数组并展平
prelab_1 = prelab_1.numpy().flatten()  # 将类别索引转换为numpy数组并展平
for i,lab in enumerate(prelab_1):  # 遍历每个预测类别及其索引print("index: {}, label: {}, probability: {}".format(lab,labels[lab][1],prob_1[i]))  # 打印每个预测类别的索引、名称和概率

index: 23, label: vulture, probability: 0.8022373914718628
index: 84, label: peacock, probability: 0.15436647832393646
index: 9, label: ostrich, probability: 0.02591262198984623
index: 88, label: macaw, probability: 0.003736903192475438
index: 17, label: jay, probability: 0.002771784085780382

从结果可以可以看到预测为vulture(秃鹫)，预测正确

VGGNet

CNN卷积网络的发展史

关键知识点

1. 简介

2. ConvNet Configurations(卷积配置)

2.1 Architecture（结构）

2.2 Configurations（网络配置）

2.3 Discussion(讨论)

3. Classification Framework（分类框架）

3.1 Training（训练细节）

3.2 Testing(测试细节)

3.3 实验

单尺度评估

多尺度评估

多裁剪评估

多模型融合

与最新模型进行比较

5. Conclusion(总结)

Pytorch实现

恭喜！

It’s not the alarm clock that wakes me up every morning,but my great idea.

每天早上叫醒我的不是闹钟而是我伟大的理想！

相关文章：