回答編集履歴
1
修正
test
CHANGED
@@ -1,6 +1,6 @@
|
|
1
1
|
(C, H, W) の正規化済みのテンソルを作りたいのであれば、以下の Transform を使えばできます。
|
2
2
|
|
3
|
-
|
3
|
+
正規化は `transforms.ToTensor()` でできます。
|
4
4
|
|
5
5
|
|
6
6
|
|
@@ -24,7 +24,7 @@
|
|
24
24
|
|
25
25
|
img_tensor = transform(img)
|
26
26
|
|
27
|
-
print(img_tensor.shape) # torch.Size([3, 256, 256])
|
27
|
+
print(img_tensor.shape) # torch.Size([3, 256, 256]) 正規化済み
|
28
28
|
|
29
29
|
```
|
30
30
|
|