update rord.py

2025-07-20 15:37:42 +08:00
parent 35c3cb1420
commit bfcd63725b
2 changed files with 4 additions and 1 deletions
--- a/models/rord.py
+++ b/models/rord.py
@@ -16,7 +16,7 @@ class RoRD(nn.Module):
        vgg16_features = models.vgg16(pretrained=False).features
        
        # 共享骨干网络 - 只使用到 relu4_3，确保特征图尺寸一致
-        self.backbone = vgg16_features[:23]  # 到 relu4_3
+        self.backbone = nn.Sequential(*list(vgg16_features.children())[:23])

        # 检测头
        self.detection_head = nn.Sequential(