可视化经典模型的对比实验总结

可视化经典模型的对比实验总结


环境安装

安装OpenGL

sudo apt-get install -y build-essential  libxmu-dev libgl1-mesa-glx libglu1-mesa-dev libgl1-mesa-dev freeglut3-dev libglew-dev libsdl2-dev libsdl2-image-dev libglm-dev libfreetype6-dev

安装Netron

网页版:https://netron.app/
官方下载:https://github.com/lutzroeder/netron

安装zetane

AlexNet

Alexnet 是一个图像分类模型,其中输入是 1000 个不同类别(例如猫、狗等)之一的图像,输出是 1000 个数字的向量。
输出向量的第i个元素是输入图像属于第i类的概率;因此,输出向量的所有元素之和为 1。
AlexNet 的输入是大小为 224x224 像素的 RGB 图像。

模型设计图

Netron结构图

weights

bias

input (1,3,224,224)

Conv

Feature Maps

ReLU

MaxPool

整体结构图

feature0

feature3

feature6

feature8

feature10

classify

VGG

模型设计图

Netron结构图

整体结构图

feature0

feature2

feature5

feature7

feature10

feature12

feature14

feature17

feature19

feature21

feature24

feature26

feature28

classifier

GoogleNet

模型设计图

Netron网络图

zetane整体结构图

详细可视化
1

2

3

分支

m_1

m_2

m_3

m_4

for_epoch 9次

output_merge

end

Inception_v3

也称为 GoogleNetv3,2015 年开始在 Imagenet 上训练的著名 ConvNet。

模型设计图

Netron结构图

zetane整体结构图

conv

分支

m_1

m_2

m_3

m_4

全局图

DenseNet

密集卷积网络(DenseNet)以前馈方式将每一层连接到其他每一层。

网络设计图

Netron网络图

这个部分可以看做是一个基础组件的结构,后面大量嵌套并循环使用

局部图

第一部分

第二部分

第三部分-头

第三部分-尾

第四部分

zetane网络图

SqueezeNet

参数减少 50 倍的 Alexnet 级精度。

网络设计图

Netron网络图

Zetene整体图

细节

features.0

features.3.squeeze

features.3.expand3x3

features.4

features.4.expand1x1

features.5.squeeze

features.5.expand3x3

features.7.squeeze

features.7.expand

features.8.squeeze

features.8.expand

features.9.squeeze

features.10.expand

features.10.squeeze

features.10.expand1x1

features.12.squeeze

features.12.expand.3x3

classifer.1

全局图

output

ShuffleNet

一个针对速度和内存进行了优化的高效 ConvNet,在 Imagenet 上进行了预训练。

所有预训练模型都期望输入图像以相同的方式归一化,即形状为 的 3 通道 RGB 图像的小批量(3 x H x W),其中HW预计至少为224。必须将图像加载到 的范围内,然后使用 和[0, 1]进行归一化。`mean = [0.485, 0.456, 0.406]``std = [0.229, 0.224, 0.225]。

网络设计图

Netron网络图

Zetane整体图

p1

这里同样可以将几个重复的结构作为单独模块来理解分析,大致可分为两个组件和一个连接件。

组件一:包含两个输出层,输出层一包含三个Conv卷积和两个ReLu,输出层二包含两个Conv卷积和一个ReLu。

组件二:包含两个输出层,输出层一无其他操作直接输出,输出层二包含三个Conv卷积和两个ReLu。

连接件:这部分主要是用来连接组件模块设计的,包含Constant、Concat、Transpose、Reshape、Split。

先来预览下组件模块的效果

组件一

组件二

连接件:

concat

input_1 (24,28,28)

input_2 (24,28,28)

output (48,28,28)

Constant

Reshape

Transpose

Split

input (48,28,28)

output_1 (24,28,28)

output_2 (24,28,28)

模型组网

组网方式一

组网方式二

end

ResNet

在 ImageNet 上预训练的深度残差网络

网络设计图

Netron网络图

Zetane整体图

局部细节

p1

p2

p3

p4

p5

p6

p7

p8

p9

p10

整体可视化

output

ResNext

下一代 ResNet,更高效、更准确

网络设计图

Netron结构图

Zetane整体图

局部细节图

下面是一个基本的结构,共包括两个输出层,其中一个输出层包含三个Conv卷积和两个ReLu,另一个输出层只包含一个Conv卷积。
最后合并两个输出通道的结果,输入下一层进行计算。

简单来看下效果

接下来是对上面基本结构的一个交叉循环,共计16个。

最终输出

output结果

Wide_ResNet

Netron网络图

Zetane整体结构

p1

两个基本结构

结构一

结构二

下面分别来看下两个结构的可视化

结构一,分为两个输出层,一层仅包含一个Conv卷积,另外一层是包含三个Conv卷积和两个ReLu。

结构二,分为两个输出层,其中一层不包含操作,另外一层与结构一的第二层一样,包含三个Conv卷积和两个ReLu。

中间连接部分,主要是承上启下,将两个输出通道的输出结果进行叠加融合,重新进行ReLu传递给下面。

RegNet

Netron网络图

Zetane整体图

两个基本结构

结构一

结构二

结构一

局部细节一

局部细节二

结构二

整体效果

局部细节

细节一

细节二

连接部分

结尾部分

EfficientNets

EfficientNets 是一系列图像分类模型,可实现最先进的精度,更小更快。使用张量核心以混合精度训练。

Netron网络设计图

Zetane整体图

局部细节

结构一

结构一 部分1

结构一 部分2

结构一 部分3

ConvNets

模型结构图

Vision_Transformer

模型结构图

参考材料:

posted @ 2022-03-05 16:33  Xu_Lin  阅读(710)  评论(0编辑  收藏  举报