修改可视化客户端ImageTk为ctk,消除警告。

2025-09-22 14:09:19 +08:00
parent 8791b4f2c0
commit 2d5096a0b2
2 changed files with 14 additions and 11 deletions
@@ -30,7 +30,8 @@ class ResNet50EmbeddingNet(nn.Module):
        
        # 加载预训练的ResNet50 (使用新的weights参数)
        if pretrained:
-            self.backbone = models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V1)
+            # V2效果会好一点
+            self.backbone = models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V2)
        else:
            self.backbone = models.resnet50(weights=None)