yang-0201/YOLOv6_pro

关于repghost

fenton-lu opened this issue · 1 comments

在repghostC3.yaml中好像只用了RepGhostC3,并没有用到repghost模块吧,RepVGGBlock只是实现了Repconv吧?
backbone:
[[-1, 1, RepVGGBlock, [64, 3, 2]], # 0-P1/2 channel: 24
[-1, 1, RepVGGBlock, [128, 3, 2]], # 1-P2/4 channel: 48
[-1, 1, RepGhostC3, [128, 192, 2, 3]],
[-1, 1, RepVGGBlock, [256, 3, 2]], # 3-P3/8 channel: 96
[-1, 1, RepGhostC3, [256, 384, 2, 5, 0.25]],
[-1, 1, RepVGGBlock, [512, 3, 2]], # 5-P4/16
[-1, 1, RepGhostC3, [512, 768, 6, 3]],
[-1, 1, RepVGGBlock, [1024, 3, 2]], # 7-P5/32
[-1, 1, RepGhostC3, [1024, 2560, 5, 5, 0.25]],
[-1, 1, SimSPPF, [1024, 5]]]

RepGhostC3中就包括了repghost模块,因为用到了RepGhostBottleneck块,而repghost就是这个模块组成的
而RepGhostC3结构就是三个1x1卷积加上一定数量的RepGhostBottleneck块。
RepVGGBlock就是RepConv,这里沿用了yolov6s中的基本主干网络格式