Update README.md
Browse files
README.md
CHANGED
@@ -107,8 +107,8 @@ transform = T.Compose([
|
|
107 |
T.Resize((width, height)),
|
108 |
T.ToTensor(), # Converts to shape [3, H, W], range [0,1]
|
109 |
T.Normalize(
|
110 |
-
mean=(0.
|
111 |
-
std=(0.
|
112 |
), # H-optimus-0 normalization
|
113 |
])
|
114 |
tile_tensor = transform(img).unsqueeze(0) # Add batch dim: [1, 3, width, height]
|
|
|
107 |
T.Resize((width, height)),
|
108 |
T.ToTensor(), # Converts to shape [3, H, W], range [0,1]
|
109 |
T.Normalize(
|
110 |
+
mean=(0.485, 0.456, 0.406),
|
111 |
+
std=(0.229, 0.224, 0.225)
|
112 |
), # H-optimus-0 normalization
|
113 |
])
|
114 |
tile_tensor = transform(img).unsqueeze(0) # Add batch dim: [1, 3, width, height]
|