yjh0410 2 年之前
父节点
当前提交
dcd32839f6
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      models/detectors/lodet/lodet_backbone.py

+ 1 - 1
models/detectors/lodet/lodet_backbone.py

@@ -21,7 +21,7 @@ class ScaleModulationNet(nn.Module):
         # P1/2
         # P1/2
         self.layer_1 = nn.Sequential(
         self.layer_1 = nn.Sequential(
             Conv(3, 16, k=3, p=1, s=2, act_type=act_type, norm_type=norm_type),
             Conv(3, 16, k=3, p=1, s=2, act_type=act_type, norm_type=norm_type),
-            Conv(16, 16, k=3, p=1, act_type=act_type, norm_type=norm_type, depthwise=depthwise),
+            Conv(16, 16, k=3, p=1, s=1, act_type=act_type, norm_type=norm_type, depthwise=depthwise),
         )
         )
 
 
         # P2/4
         # P2/4