回答編集履歴
1
修正
answer
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
(C, H, W) の正規化済みのテンソルを作りたいのであれば、以下の Transform を使えばできます。
|
|
2
|
+
正規化は `transforms.ToTensor()` でできます。
|
|
2
3
|
|
|
3
|
-
|
|
4
4
|
```python
|
|
5
5
|
from PIL import Image
|
|
6
6
|
from torchvision import transforms as transforms
|
|
@@ -11,7 +11,7 @@
|
|
|
11
11
|
[transforms.Resize((256, 256)), transforms.ToTensor()]
|
|
12
12
|
)
|
|
13
13
|
img_tensor = transform(img)
|
|
14
|
-
print(img_tensor.shape) # torch.Size([3, 256, 256])
|
|
14
|
+
print(img_tensor.shape) # torch.Size([3, 256, 256]) 正規化済み
|
|
15
15
|
```
|
|
16
16
|
|
|
17
17
|
Pytorch について詳しくないのであれば、1からコードを書くのではなく、クラス分類のサンプルや解説記事のコードをまずは動かしてみるなど、使い方を覚えたほうがいいと思います。
|