torchvision.transforms.ToTensor

1、torchvision.transforms.ToTensor把图像转化为[0,1]的torch.float32类型,并改为channel first

(torchvision.transforms.functional.to_tensor也是同样的效果)

2、torch.tensor()只是把数据类型转化为tensor,并没有改变数值范围

例如:

from torchvision.transforms import ToTensor
from PIL import Image
import numpy as np

img = Image.open()
print(img)

>>> 
#这个size是width*high

img2 = np.array(img)
print(img2.shape)
>>> (300, 400, 3) #high在第一个位置

img3 = ToTensor()(img2)#转化为channel first
print(img3.shape)
print(img3.dtype)

>>>torch.Size([3, 300, 400])
>>>torch.float32


img4 = torch.tensor(img2)
print(img4.shape)

>>> torch.Size([300, 400, 3])

你可能感兴趣的:(pytorch,深度学习,python)