目录
一、原理
其实就是借鉴了这个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([