Update yolov9-c-hyper.yaml

iMoonLab · Aug 9, 2024 · 2e47a42 · 2e47a42
1 parent 41fc416
commit 2e47a42
Showing 1 changed file with 38 additions and 38 deletions.
diff --git a/models/detect/yolov9-c-hyper.yaml b/models/detect/yolov9-c-hyper.yaml
@@ -47,73 +47,73 @@ backbone:
 head:
   [
    # elan-spp block
-   [-1, 1, SPPELAN, [512, 256]],  # 9+1
+   [-1, 1, SPPELAN, [512, 256]],  # 10
 
-   [1, 1, nn.AvgPool2d, [8, 8, 0]], # 10+1
-   [3, 1, nn.AvgPool2d, [4, 4, 0]], # 11+1
-   [5, 1, nn.AvgPool2d, [2, 2, 0]], # 12+1
-   [10, 1, nn.Upsample, [None, 2, 'nearest']], # 13+1
-   [[11, 12, 13, 7, 14], 1, Concat, [1]], # 14+1
-   [-1, 1, Conv, [512, 1, 1]], # 15+1
-   [-1, 1, HyperComputeModule, [512]], # 16+1
-   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 17+1
+   [1, 1, nn.AvgPool2d, [8, 8, 0]], # 11
+   [3, 1, nn.AvgPool2d, [4, 4, 0]], # 12
+   [5, 1, nn.AvgPool2d, [2, 2, 0]], # 13
+   [10, 1, nn.Upsample, [None, 2, 'nearest']], # 14
+   [[11, 12, 13, 7, 14], 1, Concat, [1]], # 15
+   [-1, 1, Conv, [512, 1, 1]], # 16
+   [-1, 1, HyperComputeModule, [512]], # 17
+   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 18
 
-   [-1, 1, nn.AvgPool2d, [2, 2, 0]], # 18+1
-   [[-1, 10], 1, Concat, [ 1 ]], # 19+1
-   [-1, 1, Conv, [512, 1, 1]], # 20+1
+   [-1, 1, nn.AvgPool2d, [2, 2, 0]], # 19
+   [[-1, 10], 1, Concat, [ 1 ]], # 20
+   [-1, 1, Conv, [512, 1, 1]], # 21
 
-   [[18, 7], 1, Concat, [1]], # 21+1
-   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 22+1
+   [[18, 7], 1, Concat, [1]], # 22
+   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 23
 
-   [18, 1, nn.Upsample, [None, 2, 'nearest']], # 23+1
-   [[-1, 5], 1, Concat, [1]], # 24+1
-   [-1, 1, RepNCSPELAN4, [256, 256, 128, 1]], # 25+1 
+   [18, 1, nn.Upsample, [None, 2, 'nearest']], # 24
+   [[-1, 5], 1, Concat, [1]], # 25
+   [-1, 1, RepNCSPELAN4, [256, 256, 128, 1]], # 26
 
-   [-1, 1, ADown, [256]], # 26+1
-   [[-1, 23], 1, Concat, [1]], # 27+1
-   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 28+1 
+   [-1, 1, ADown, [256]], # 27
+   [[-1, 23], 1, Concat, [1]], # 28
+   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 29
 
-   [-1, 1, ADown, [512]], # 29+1
-   [[-1, 21], 1, Concat, [1]], # 30+1
-   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 31+1 
+   [-1, 1, ADown, [512]], # 30
+   [[-1, 21], 1, Concat, [1]], # 31
+   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # 32
 
 
    # multi-level reversible auxiliary branch
 
    # routing
-   [5, 1, CBLinear, [[256]]], # 23+10
-   [7, 1, CBLinear, [[256, 512]]], # 24+10
-   [9, 1, CBLinear, [[256, 512, 512]]], # 25+10
+   [5, 1, CBLinear, [[256]]], # 33
+   [7, 1, CBLinear, [[256, 512]]], # 34
+   [9, 1, CBLinear, [[256, 512, 512]]], # 35
 
    # conv down
-   [0, 1, Conv, [64, 3, 2]],  # 26+10-P1/2
+   [0, 1, Conv, [64, 3, 2]],  # 36-P1/2
 
    # conv down
-   [-1, 1, Conv, [128, 3, 2]],  # 27+10-P2/4
+   [-1, 1, Conv, [128, 3, 2]],  # 37-P2/4
 
    # elan-1 block
-   [-1, 1, RepNCSPELAN4, [256, 128, 64, 1]],  # 28+10
+   [-1, 1, RepNCSPELAN4, [256, 128, 64, 1]],  # 38
 
    # avg-conv down fuse
-   [-1, 1, ADown, [256]],  # 29+10-P3/8
-   [[33, 34, 35, -1], 1, CBFuse, [[0, 0, 0]]], # 30+10
+   [-1, 1, ADown, [256]],  # 39-P3/8
+   [[33, 34, 35, -1], 1, CBFuse, [[0, 0, 0]]], # 40
 
    # elan-2 block
-   [-1, 1, RepNCSPELAN4, [512, 256, 128, 1]],  # 31+10
+   [-1, 1, RepNCSPELAN4, [512, 256, 128, 1]],  # 41
 
    # avg-conv down fuse
-   [-1, 1, ADown, [512]],  # 32+10-P4/16
-   [[34, 35, -1], 1, CBFuse, [[1, 1]]], # 33+10
+   [-1, 1, ADown, [512]],  # 42-P4/16
+   [[34, 35, -1], 1, CBFuse, [[1, 1]]], # 43
 
    # elan-2 block
-   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]],  # 34+10
+   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]],  # 44
 
    # avg-conv down fuse
-   [-1, 1, ADown, [512]],  # 35+10-P5/32
-   [[35, -1], 1, CBFuse, [[2]]], # 36+10
+   [-1, 1, ADown, [512]],  # 45-P5/32
+   [[35, -1], 1, CBFuse, [[2]]], # 46
 
    # elan-2 block
-   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]],  # 37+10
+   [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]],  # 47