当前位置:   article > 正文

复现Underwater target detection based on improved YOLOv7(基于改进YOLOV7的水下目标检测

复现Underwater target detection based on improved YOLOv7(基于改进YOLOV7的水下目标检测

主要改进点:

(1)为了提取信息更丰富的特征,提出了整合全局注意力机制(GAM)。 该机制有效地捕获了特征的通道和空间方面,并增加了跨维交互的重要性。

(2)为了进一步提高网络的性能,引入了ACMIX(一种结合自关注和卷积优点的混合模型)。

(3)Yolov7-AC中ResNet-ACMIX模块的设计旨在增强骨干网的特征提取能力,并通过捕获更多信息特征来加速网络的收敛。

(4)对Yolov7网络中的E-ELAN模块进行了优化,在模块间引入了跳过连接和1×1卷积结构,并用ACMIXBLOCK取代了3×3卷积层。 这提高了特征提取能力,提高了推理速度。

ResNet-ACmix module:

 

在Yolov7的主干组件中引入ResNet-ACMIX模块,有效地保持了提取特征信息的一致性。 该模块基于ResNet的瓶颈结构,其中3x3卷积被ACMIX模块所取代,从而能够对不同区域进行自适应聚焦,并捕获更多信息特征,如图所示。 将输入分为主输入和残差输入,这有助于防止信息丢失,同时减少了参数数量和计算要求。 RESNet-ACMIX模块使网络能够在不遇到梯度消失的情况下获得更深的深度,并且学习结果对网络权值的波动更敏感。

 ACmixblock代码如下:

  1. class ACmixblock(nn.Module):
  2. # Standard convolution
  3. def __init__(self, in_planes, out_planes, k=1, s=1, p=None, g=1, act=True): # ch_in, ch_out, kernel, stride, padding, groups
  4. super(ACmixblock, self).__init__()
  5. self.conv = ACmix(in_planes, out_planes, kernel_att=7, head=4, kernel_conv=3, stride=1, dilation=1)
  6. self.bn = nn.BatchNorm2d(out_planes)
  7. self.act = nn.ReLU(inplace=True) if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
  8. def forward(self, x):
  9. return self.act(self.bn(self.conv(x)))
  10. def fuseforward(self, x):
  11. return self.act(self.conv(x))

 ResNet-ACmix代码如下:

  1. #ResNet-ACmix
  2. class ResNet_ACmix(nn.Module):
  3. # ResNet bottleneck
  4. def __init__(self, c1, c2, shortcut=True, g=1, e=0.5): # ch_in, ch_out, shortcut, groups, expansion
  5. super(ResNet_ACmix, self).__init__()
  6. c_ = int(c2 * e) # hidden channels
  7. self.cv1 = Conv(c1, c_, 1, 1)
  8. self.cv2 = ACmixblock(c_, c_, kernel_att=7, head=4, kernel_conv=3, stride=1, dilation=1)
  9. self.cv3 = Conv(c_, c2, 1, 1)
  10. self.add = shortcut and c1 == c2
  11. def forward(self, x):
  12. return x + self.cv3(self.cv2(self.cv1(x))) if self.add else self.cv3(self.cv2(self.cv1(x)))

AC-E-ELAN module

Yolov7中E-ELAN组件的特征提取模块通过加入REPVGG结构中的残差结构(即1×1卷积分支和跳转连接分支)得到了进一步的改进。 这导致了AC-E-ELAN结构的发展,如图所示,它集成了ACMIXBlock,由3x3卷积块组成,在ACMIXBlocks之间具有跳转连接和1x1卷积结构。 这种组合使网络既能从多分支模型训练过程中获得的丰富特征中获益,又能从单路径模型中获得快速、有效的记忆推断。

  1. # AC-E-ELAN
  2. class RepACmixblock(nn.Module):
  3. def __init__(self, c1, c2, k=3, s=1, p=None, g=1, act=True, deploy=False,kernel_att=7, head=4, dilation=1):
  4. super(RepACmixblock, self).__init__()
  5. self.deploy = deploy
  6. self.groups = g
  7. self.in_channels = c1
  8. self.out_channels = c2
  9. self.act = nn.ReLU() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
  10. if deploy:
  11. self.rbr_reparam = ACmixblock(c1, c2)
  12. else:
  13. self.rbr_identity = (nn.BatchNorm2d(num_features=c1) if c2 == c1 and s == 1 else None)
  14. self.rbr_dense = ACmixblock(c1, c2)
  15. self.rbr_1x1 = nn.Sequential(
  16. ACmix( c1, c2),
  17. nn.BatchNorm2d(num_features=c2),
  18. )
  19. def forward(self, inputs):
  20. if hasattr(self, "rbr_reparam"):
  21. return self.rbr_reparam(inputs)
  22. if self.rbr_identity is None:
  23. id_out = 0
  24. else:
  25. id_out = self.rbr_identity(inputs)
  26. return self.rbr_dense(inputs) + self.act(self.rbr_1x1(inputs) + id_out)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/51989
推荐阅读
相关标签
  

闽ICP备14008679号