yjh0410 1 år sedan
förälder
incheckning
c74363671a
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      models/yolov8/yolov8_backbone.py

+ 1 - 1
models/yolov8/yolov8_backbone.py

@@ -22,7 +22,7 @@ class Yolov8Backbone(nn.Module):
         # ------------------ Network setting ------------------
         ## P1/2
         self.layer_1 = BasicConv(3, self.feat_dims[0],
-                                 kernel_size=3, padding=1, stride=2,
+                                 kernel_size=6, padding=2, stride=2,
                                  act_type=cfg.bk_act, norm_type=cfg.bk_norm, depthwise=cfg.bk_depthwise)
         # P2/4
         self.layer_2 = nn.Sequential(