Update hyper-yolot.yaml

iMoonLab · Aug 9, 2024 · 16071e0 · 16071e0
1 parent 6021580
commit 16071e0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ultralytics/cfg/models/hyper-yolo/hyper-yolot.yaml b/ultralytics/cfg/models/hyper-yolo/hyper-yolot.yaml
@@ -21,15 +21,19 @@ backbone:
 
 # head
 head:
+ # Semantic Collecting
  - [0, 1, nn.AvgPool2d, [8, 8, 0]] # 10
  - [2, 1, nn.AvgPool2d, [4, 4, 0]] # 11
  - [4, 1, nn.AvgPool2d, [2, 2, 0]] # 12
  - [9, 1, nn.Upsample, [None, 2, 'nearest']] # 13
  - [[10, 11, 12, 6, 13], 1, Concat, [1]]  # cat 14
+
+ # Hypergraph Computation
  - [-1, 1, Conv, [512, 1, 1]] # 15
  - [-1, 1, HyperComputeModule, [512]] # 16
  - [-1, 3, MANet, [512, True]] # 17
 
+ # Semantic Scattering
  - [-1, 1, nn.AvgPool2d, [2, 2, 0]] # 18
  - [[-1, 9], 1, Concat, [1]]  # cat 19
  - [-1, 1, Conv, [1024, 1, 1]] # 20