yolov5的改进策略（YOLOv5改进之YOLOv5+GSConv+Slim Neck）

论文题目：Slim-neck by GSConv: A better design paradigm of detector architectures for autonomous vehicles

论文：https://arxiv.org/abs/2206.02424

代码：https://github.com/AlanLi1997/Slim-neck-by-GSConv

直接步入正题~~~

目标：为YOLOv5模型构建一个简单高效的Neck模块 
  
  
  
  
  。考虑了卷积方法           、特征融合结构 
  
 
  
 
  
 
  
 
  
 
、计算效率










、计算成本效益等诸多因素
  
  
  
  
  
 。

一 
  
  
  
  
  、GSConv

class GSConv(nn.Module): # GSConv https://github.com/AlanLi1997/slim-neck-by-gsconv def __init__(self, c1, c2, k=1, s=1, g=1, act=True): super().__init__() c_ = c2 // 2 self.cv1 = Conv(c1, c_, k, s, None, g, act) self.cv2 = Conv(c_, c_, 5, 1, None, c_, act) def forward(self, x): x1 = self.cv1(x) x2 = torch.cat((x1, self.cv2(x1)), 1) # shuffle b, n, h, w = x2.data.size() b_n = b * n // 2 y = x2.reshape(b_n, 2, h * w) y = y.permute(1, 0, 2) y = y.reshape(2, -1, n // 2, h, w) return torch.cat((y[0], y[1]), 1)

将YOLOv5s.yaml的Neck模块中的Conv换成GSConv

1 
  
  
  
  
  
、将GSConv代码加入common.py文件中

2

 


 


 


 


 

、找到yolo.py文件里的parse_model函数           ，将类名加入进去

3 
 
 
 
 
 、修改配置文件 
  
 
  
 
  
 
  
 
  
 
，将YOLOv5s.yaml的Neck模块中的Conv换成GSConv 

~~~此处有一个疑问










，官方给出的GSConv代码中为什么没用DWConv呢？希望知道的朋友在评论区指点一下~~~

二 
   
   
   
   
   
、GSConv+Slim Neck

1
 

 

 

 

 
、GSBottleneck

class GSBottleneck(nn.Module): # GS Bottleneck https://github.com/AlanLi1997/slim-neck-by-gsconv def __init__(self, c1, c2, k=3, s=1): super().__init__() c_ = c2 // 2 # for lighting self.conv_lighting = nn.Sequential( GSConv(c1, c_, 1, 1), GSConv(c_, c2, 1, 1, act=False)) # for receptive field self.conv = nn.Sequential( GSConv(c1, c_, 3, 1), GSConv(c_, c2, 3, 1, act=False)) self.shortcut = Conv(c1, c2, 3, 1, act=False) def forward(self, x): return self.conv_lighting(x) + self.shortcut(x)

2
 

 

 

 

 

 、VoVGSCSP

class VoVGSCSP(nn.Module): # VoV-GSCSP https://github.com/AlanLi1997/slim-neck-by-gsconv def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): super().__init__() c_ = int(c2 * e) self.cv1 = Conv(c1, c_, 1, 1) self.cv2 = Conv(2 * c_, c2, 1) self.m = nn.Sequential(*(GSBottleneck(c_, c_) for _ in range(n))) def forward(self, x): x1 = self.cv1(x) return self.cv2(torch.cat((self.m(x1), x1), dim=1))

将YOLOv5s.yaml的Neck模块中的Conv换成GSConv 
  
  
  
  
  ，C3模块换为VoVGSCSP模块

1          、将以下代码加入common.py文件中

class GSConv(nn.Module): # GSConv https://github.com/AlanLi1997/slim-neck-by-gsconv def __init__(self, c1, c2, k=1, s=1, g=1, act=True): super().__init__() c_ = c2 // 2 self.cv1 = Conv(c1, c_, k, s, None, g, act) self.cv2 = Conv(c_, c_, 5, 1, None, c_, act) def forward(self, x): x1 = self.cv1(x) x2 = torch.cat((x1, self.cv2(x1)), 1) # shuffle b, n, h, w = x2.data.size() b_n = b * n // 2 y = x2.reshape(b_n, 2, h * w) y = y.permute(1, 0, 2) y = y.reshape(2, -1, n // 2, h, w) return torch.cat((y[0], y[1]), 1) class GSBottleneck(nn.Module): # GS Bottleneck https://github.com/AlanLi1997/slim-neck-by-gsconv def __init__(self, c1, c2, k=3, s=1): super().__init__() c_ = c2 // 2 # for lighting self.conv_lighting = nn.Sequential( GSConv(c1, c_, 1, 1), GSConv(c_, c2, 1, 1, act=False)) # for receptive field self.conv = nn.Sequential( GSConv(c1, c_, 3, 1), GSConv(c_, c2, 3, 1, act=False)) self.shortcut = nn.Identity() def forward(self, x): return self.conv_lighting(x) class VoVGSCSP(nn.Module): # VoV-GSCSP https://github.com/AlanLi1997/slim-neck-by-gsconv def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): super().__init__() c_ = int(c2 * e) self.cv1 = Conv(c1, c_, 1, 1) self.cv2 = Conv(2 * c_, c2, 1) self.m = nn.Sequential(*(GSBottleneck(c_, c_) for _ in range(n))) def forward(self, x): x1 = self.cv1(x) return self.cv2(torch.cat((self.m(x1), x1), dim=1))

2 

 
 

 
 

 
 

 
 

 
 
、找到yolo.py文件里的parse_model函数 
  
  
  
  
  
，将类名加入进去

 


 


 


 


 

，注意有两处需要添加的地方

3










、修改配置文件 
 
 
 
 
 ，将YOLOv5s.yaml的Neck模块中的Conv换成GSConv  
   
   
   
   
   
，C3模块换为VoVGSCSP

Appendix

下图是原论文中给出的结构图
 

 

 

 

 
，个人对照源码后觉得这里多画了一个GSConv模块(红色框里所示)
 

 

 

 

 

 ，如果有知道的大佬望在评论区指点一下
 


 


 


 


 


。