yolov5:Conv类参数量计算

Conv是yolov5自定义的类,里边包含了卷积层、BN层和激活函数

class Conv(nn.Module):
    # Standard convolution with args(ch_in, ch_out, kernel, stride, padding, groups, dilation, activation)
    default_act = nn.SiLU()  # default activation

    def __init__(self, c1, c2, k=1, s=1, p=None, g=1, d=1, act=True):
        """Initializes a standard convolution layer with optional batch normalization and activation."""
        super().__init__()
        self.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p, d), groups=g, dilation=d, bias=False)
        self.bn = nn.BatchNorm2d(c2)
        self.act = self.default_act if act is True else act if isinstance(act, nn.Module) else nn.Identity()

假设输入是[3, 32, 6, 2, 2](就是yolov5s第一层),即输入通道=3,输出通道=32,kernel_size=3,padding=2,stride=2

于是有:
参数量=conv2d参数+BN2d参数=3*6*6*32+32*2 = 3520

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-10 00:40:05       66 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-10 00:40:05       70 阅读
  3. 在Django里面运行非项目文件

    2024-07-10 00:40:05       57 阅读
  4. Python语言-面向对象

    2024-07-10 00:40:05       68 阅读

热门阅读

  1. GitHub每日最火火火项目(7.9)

    2024-07-10 00:40:05       22 阅读
  2. py基础语法简述

    2024-07-10 00:40:05       19 阅读
  3. 代码随想录算法训练营:20/60

    2024-07-10 00:40:05       21 阅读
  4. 【6-1:全链路压测】

    2024-07-10 00:40:05       22 阅读
  5. 识别色带后执行相应命令

    2024-07-10 00:40:05       23 阅读
  6. QMdiAreaQMdiAreaQMdiAreaQMdiArea

    2024-07-10 00:40:05       22 阅读
  7. Jacoco的覆盖率原理

    2024-07-10 00:40:05       20 阅读
  8. 中英双语介绍美国的州:阿肯色州(Arkansas)

    2024-07-10 00:40:05       28 阅读
  9. Socket网络通信流程

    2024-07-10 00:40:05       27 阅读
  10. 《妃梦千年》第二十九章:朝中波澜

    2024-07-10 00:40:05       22 阅读