decoupled_light_FGEM

models/yolo.py

@@ -48,6 +48,7 @@ from models.common import (
     GhostBottleneck,
     GhostConv,
     Proto,
+    FGEMLite,   # 新增
 )
 from models.experimental import MixConv2d
 from utils.autoanchor import check_anchor_order
@@ -152,7 +153,14 @@ class DetectDecoupledLite(nn.Module):
         self.regobj_convs = nn.ModuleList(Conv(head_c, head_c, 3, 1) for _ in ch)
 
         # cls trunk
-        self.cls_convs = nn.ModuleList(Conv(head_c, head_c, 3, 1) for _ in ch)
+        # self.cls_convs = nn.ModuleList(Conv(head_c, head_c, 3, 1) for _ in ch)
+        # cls trunk + FGEMLite
+        self.cls_convs = nn.ModuleList(
+            nn.Sequential(
+                Conv(head_c, head_c, 3, 1),
+                FGEMLite(head_c, head_c // 2)
+            ) for _ in ch
+        )
 
         # predictors
         self.box_preds = nn.ModuleList(nn.Conv2d(head_c, self.na * 4, 1) for _ in ch)