From bfcd63725b37bd7dc9a0d5bcfe661abf5772d7ec Mon Sep 17 00:00:00 2001
From: Jiao77 <jiao.ts.zju@gmail.com>
Date: Sun, 20 Jul 2025 15:37:42 +0800
Subject: [PATCH] update rord.py

---
 .gitignore     | 3 +++
 models/rord.py | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 505a3b1..ccbe279 100644
--- a/.gitignore
+++ b/.gitignore
@@ -8,3 +8,6 @@ wheels/
 
 # Virtual environments
 .venv
+
+# Model Path File
+model.path/
\ No newline at end of file
diff --git a/models/rord.py b/models/rord.py
index d9dd515..2d97705 100644
--- 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(