增加TODO
This commit is contained in:
@@ -56,6 +56,7 @@ class ResNet50EmbeddingNet(nn.Module):
|
||||
)
|
||||
|
||||
# 图片预处理变换(用于推理)
|
||||
# TODO 这里应该要修改归一化的逻辑,不然不一样。
|
||||
self.preprocess = transforms.Compose([
|
||||
transforms.Resize((224, 224), interpolation=transforms.InterpolationMode.BILINEAR),
|
||||
transforms.ToTensor(),
|
||||
|
||||
Reference in New Issue
Block a user