yolov5 CSP结构

YOLOv5中的CSP构是指Cross Stage Partial Network，是一种用于目标检测的卷积神经网络结构。它的设计旨在提高模型的准确性和速度。YOLOv5中的CSP结构有两种设计，分别为CSP1_X结构和CSP2_X结构。

其中，CSP1_X结构是将输入先进行一次卷积操作，然后将输出分成两个分支，其中一个分支进行BottleneckCSP操作，另一个分支直接输出。最后将两个分支的输出进行concat操作，得到最终的输出。

而CSP2_X结构则是在CSP1_X结构的基础上增加了一个SPP结构，即Spatial Pyramid Pooling，用于提取不同尺度的特征。

以YOLOv5s为例，它的CSP结构是将原输入分成两个分支，分别进行卷积操作使得通道数减半，然后一个分支进行Bottleneck *N操作，然后concat两个分支，使得BottlenneckCSP的输入与输出是一样的大小，这样是为了让模型学习到更多的特征。

# YOLOv5 CSP结构示例代码
import torch.nn as nn

class CSPBlock(nn.Module):
    def __init__(self, in_channels, out_channels, n=1):
        super(CSPBlock, self).__init__()
        self.conv1 = nn.Conv2d(in_channels, out_channels, 1, 1, bias=False)
        self.conv2 = nn.Conv2d(in_channels, out_channels, 1, 1, bias=False)
        self.conv3 = nn.Conv2d(out_channels, out_channels, 1, 1, bias=False)
        self.bn = nn.BatchNorm2d(out_channels * 2)
        self.relu = nn.LeakyReLU(0.1, inplace=True)
        self.bottleneck = nn.Sequential(
            *[Bottleneck(out_channels, out_channels // 2) for _ in range(n)]
        )

    def forward(self, x):
        y1 = self.conv1(x)
        y2 = self.conv2(x)
        y2 = self.bottleneck(y2)
        y = torch.cat([y1, y2], dim=1)
        y = self.conv3(y)
        y = self.bn(y)
        y = self.relu(y)
        return y