增加TODO

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