test

src/dataset/preprocess.py

@@ -12,3 +12,9 @@ def make_transform(resize_size: int = 256):
         std=(0.229, 0.224, 0.225),
     )
     return v2.Compose([to_tensor, resize, to_float, normalize])
+
+
+def denormalize(tensor):
+    mean = torch.tensor([0.485, 0.456, 0.406]).view(3, 1, 1).to(tensor.device)
+    std = torch.tensor([0.229, 0.224, 0.225]).view(3, 1, 1).to(tensor.device)
+    return tensor * std + mean