yjh0410 2 years ago
parent
commit
43562b385a
1 changed files with 1 additions and 1 deletions
  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])
         self.encoder = build_neck(cfg, feats_dim[-1], feats_dim[-1])
 
 
         ## CSFM
         ## 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):
     def forward(self, x):