From 8791b4f2c0ec12ac184a7313cf617049c13cc52e Mon Sep 17 00:00:00 2001
From: zhanghuan <1262329256@qq.com>
Date: Mon, 22 Sep 2025 13:59:05 +0800
Subject: [PATCH] =?UTF-8?q?pretrained=3Dpretrained,=E8=A7=A3=E5=86=B3?=
 =?UTF-8?q?=E9=A2=84=E8=AE=AD=E7=BB=83=E6=A8=A1=E5=9E=8BAPI=E5=8F=98?=
 =?UTF-8?q?=E5=8C=96=E5=AF=BC=E8=87=B4=E7=9A=84=E8=AD=A6=E5=91=8A=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 net/resnet_embedding.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/net/resnet_embedding.py b/net/resnet_embedding.py
index 9f8175d..7d8c952 100644
--- a/net/resnet_embedding.py
+++ b/net/resnet_embedding.py
@@ -28,8 +28,11 @@ class ResNet50EmbeddingNet(nn.Module):
         self.embedding_dim = embedding_dim
         self.use_internal_preprocess = use_internal_preprocess
         
-        # 加载预训练的ResNet50
-        self.backbone = models.resnet50(pretrained=pretrained)
+        # 加载预训练的ResNet50 (使用新的weights参数)
+        if pretrained:
+            self.backbone = models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V1)
+        else:
+            self.backbone = models.resnet50(weights=None)
         
         # 获取ResNet50最后一层的输入特征数（2048）
         backbone_output_dim = self.backbone.fc.in_features