YOLO系列改进，自研模块助力涨点

目录

一、原理

二、代码

三、添加到YOLOv5中

---

一、原理

论文地址：2103.13425 (arxiv.org)

其实就是借鉴了这个DBB模块的重参数化的思想，简化了DBB模块的程序

简化之后的模块结构，跟DBB类似。

二、代码

class DiverseBranchBlock(nn.Module):
    def __init__(self, in_channels, out_channels, kernel_size=3):
        super(DiverseBranchBlock, self).__init__()

        inp = out_channels//4

        self.branches = nn.ModuleList([