improved rf

2025-10-01 18:44:26 +09:00
parent 49025c4d87
commit 29eb04d1a4
8 changed files with 150 additions and 27 deletions
--- a/src/dataset/preprocess.py
+++ b/src/dataset/preprocess.py
@@ -14,7 +14,7 @@ def make_transform(resize_size: int = 256):
    return v2.Compose([to_tensor, resize, to_float, normalize])


-def denormalize(tensor):
+def denormalize(tensor: torch.Tensor) -> torch.Tensor:
    mean = torch.tensor([0.430, 0.411, 0.296]).view(3, 1, 1).to(tensor.device)
    std = torch.tensor([0.213, 0.156, 0.143]).view(3, 1, 1).to(tensor.device)
    return tensor * std + mean