增加TODO

2025-10-09 18:05:43 +08:00
parent 7faf3f6480
commit 86a4f1ff67
1 changed files with 1 additions and 0 deletions
@@ -56,6 +56,7 @@ class ResNet50EmbeddingNet(nn.Module):
        )
        # 图片预处理变换（用于推理）
        # TODO 这里应该要修改归一化的逻辑，不然不一样。
        self.preprocess = transforms.Compose([
            transforms.Resize((224, 224), interpolation=transforms.InterpolationMode.BILINEAR),
            transforms.ToTensor(),