yjh0410 %!s(int64=2) %!d(string=hai) anos
pai
achega
43562b385a
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      models/detectors/rtdetr/image_encoder/img_encoder.py

+ 1 - 1
models/detectors/rtdetr/image_encoder/img_encoder.py

@@ -17,7 +17,7 @@ class ImageEncoder(nn.Module):
         self.encoder = build_neck(cfg, feats_dim[-1], feats_dim[-1])
 
         ## CSFM
-        self.csfm = build_fpn(cfg=cfg, in_dims=feats_dim, out_dim=round(cfg['d_model']*cfg['width']))
+        self.csfm = build_fpn(cfg=cfg, in_dims=feats_dim, out_dim=round(cfg['d_model']*cfg['width']), input_proj=True)
 
 
     def forward(self, x):