Bladeren bron

modify MDown

yjh0410 1 jaar geleden
bovenliggende
commit
f99d38377c
1 gewijzigde bestanden met toevoegingen van 1 en 7 verwijderingen
  1. 1 7
      yolo/models/yolov7_af/yolov7_af_basic.py

+ 1 - 7
yolo/models/yolov7_af/yolov7_af_basic.py

@@ -90,18 +90,12 @@ class MDown(nn.Module):
                       kernel_size=3, padding=1, stride=2,
                       act_type=act_type, norm_type=norm_type, depthwise=depthwise)
         )
-        if in_dim == out_dim:
-            self.output_proj = nn.Identity()
-        else:
-            self.output_proj = BasicConv(inter_dim * 2, out_dim, kernel_size=1, act_type=act_type, norm_type=norm_type)
 
     def forward(self, x):
         x1 = self.downsample_1(x)
         x2 = self.downsample_2(x)
 
-        out = self.output_proj(torch.cat([x1, x2], dim=1))
-
-        return out
+        return torch.cat([x1, x2], dim=1)
 
 class ELANLayer(nn.Module):
     def __init__(self,