admin 管理员组

文章数量: 887021


2024年2月20日发(作者:sqrt4840000)

darknet53 卷积参数量计算

对于DarkNet-53的卷积参数量计算,我们首先需要知道该网络的结构和每个卷积层的参数设置。

DarkNet-53网络结构如下:

1. 7x7卷积层,64个卷积核,步长为2,padding为3。

2. 残差块1:1x1卷积层,32个卷积核,步长为1,padding为0;3x3卷积层,64个卷积核,步长为1,padding为1。

3. 残差块2:1x1卷积层,32个卷积核,步长为1,padding为0;3x3卷积层,64个卷积核,步长为2,padding为1。

4. 残差块3:1x1卷积层,64个卷积核,步长为1,padding为0;3x3卷积层,128个卷积核,步长为2,padding为1。

5. 残差块4:1x1卷积层,128个卷积核,步长为1,padding为0;3x3卷积层,256个卷积核,步长为2,padding为1。

6. 残差块5:1x1卷积层,256个卷积核,步长为1,padding为0;3x3卷积层,512个卷积核,步长为2,padding为1。

7. 残差块6:1x1卷积层,512个卷积核,步长为1,padding为0;3x3卷积层,1024个卷积核,步长为2,padding为1。

8. 残差块7:1x1卷积层,1024个卷积核,步长为1,padding为0;3x3卷积层,2048个卷积核,步长为1,padding为1。

9. 全局平均池化层。

接下来,我们计算卷积参数量:

1. 第一层7x7卷积:输入通道数为3,每个卷积核大小为7x7x3,共64个卷积核,所以参数量为7x7x3x64 = 9408。

2. 残差块1:1x1卷积参数量为1x1x64x32 = 2048;3x3卷积参数量为3x3x32x64 = 18432。故总参数量为2048 + 18432 = 20480。

3. 残差块2:1x1卷积参数量为1x1x64x32 = 2048;3x3卷积参数量

为3x3x32x64 = 18432。故总参数量为2048 + 18432 = 20480。

4. 残差块3:1x1卷积参数量为1x1x128x64 = 8192;3x3卷积参数量为3x3x64x128 = 73728。故总参数量为8192 + 73728 = 81920。

5. 残差块4:1x1卷积参数量为1x1x256x128 = 32768;3x3卷积参数量为3x3x128x256 = 294912。故总参数量为32768 + 294912 =

327680。

6. 残差块5:1x1卷积参数量为1x1x512x256 = 131072;3x3卷积参数量为3x3x256x512 = 1179648。故总参数量为131072 + 1179648 =

1310720。

7. 残差块6:1x1卷积参数量为1x1x1024x512 = 524288;3x3卷积参数量为3x3x512x1024 = 4718592。故总参数量为524288 + 4718592 =

5242880。

8. 残差块7:1x1卷积参数量为1x1x2048x1024 = 2097152;3x3卷积参数量为3x3x1024x2048 = 18874368。故总参数量为2097152 +

18874368 = 20971520。

9. 全局平均池化层:无参数,不计入总参数量。

最终DarkNet-53的卷积参数量为:9408 + 20480 + 20480 + 81920 +

327680 + 1310720 + 5242880 + 20971520 = 81285168。


本文标签: 卷积 数量 步长 总参 网络