|
|
@@ -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(
|
yjh0410