YOLOv8的c2f模块详解

最新推荐文章于 2025-03-08 11:23:16 发布

下饭程序员

最新推荐文章于 2025-03-08 11:23:16 发布

阅读量2w

点赞数 25

文章标签： YOLO python 开发语言

本文链接：https://blog.csdn.net/weixin_43227262/article/details/130209037

版权

文章详细介绍了YOLOv8网络架构中的C2f模块，该模块基于CSPBottleneck设计，包含了两个卷积层和多个Bottleneck块。Bottleneck块是标准的瓶颈结构，用于提升模型性能。C2f模块通过分割输入并应用注意力机制，增强了特征表示能力。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

C2f模块的代码在YOLOv8上可以找到

class C2f(nn.Module):
    """CSP Bottleneck with 2 convolutions."""

    def __init__(self, c1, c2, n=1, shortcut=False, g=1, e=0.5):  # ch_in, ch_out, number, shortcut, groups, expansion
        super().__init__()
        self.c = int(c2 * e)  # hidden channels
        self.cv1 = Conv(c1, 2 * self.c, 1, 1)
        self.cv2 = Conv((2 + n) * self.c, c2, 1)  # optional act=FReLU(c2)
        self.m = nn.ModuleList(Bottleneck(self.c, self.c, shortcut, g, k=((3, 3), (3, 3)), e=1.0) for _ in range(n))

    def forward(self, x):
        """Forward pass of a YOLOv5 CSPDarknet backbone layer."""
        y = list(self.cv1(x).chunk(2, 1))
        y.extend(m(y[-1]) for m in self.m)
        return self.cv2(torch.cat(y, 1))

    def forward_split(self, x):
        """Applies spatial attention to module's input."""
        y = list(self.cv1(x).split((self.c, self.c), 1))
        y.extend(m(y[-1]) for m in self.m)
        return self.cv2(torch.cat(y, 1))

里面包含Bottleneck

class Bottleneck(nn.Module):
    """Standard bottleneck."""

    def __init__(self, c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5):  # ch_in, ch_out, shortcut, groups, kernels, expand
        super().__init__()
        c_ = int(c2 * e)  # hidden channels
        self.cv1 = Conv(c1, c_, k[0], 1)
        self.cv2 = Conv(c_, c2, k[1], 1, g=g)
        self.add = shortcut and c1 == c2

    def forward(self, x):
        """'forward()' applies the YOLOv5 FPN to input data."""
        return x + self.cv2(self.cv1(x)) if self.add else self.cv2(self.cv1(x))