yjh0410 преди 2 години
родител
ревизия
43562b385a
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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):