How-PIL-to-Tensor

图片是如何从$NumPy$形式向$Tensor$形式转换的呢?

from PIL import Image
import numpy as np
from torchvision.transforms import ToTensor
img=Image.open('./sword.png')
img_arr=np.array(img)
img_tensor=ToTensor().__call__(img)

$NumPy$的图片格式$H,W,C$，即$height,width,channels$(高、宽、通道数)

img_arr.shape

(73, 183, 4)

$Tensor$图片格式为$C,H,W$，即$channels,height,width$(通道数、高、宽)

img_tensor.shape

torch.Size([4, 73, 183])

实际上将$NumPy$数组每个位置上对应的像素值除以$255$就得到了对应的$Tensor$数据类型

# 取出图像左上角第一个点的RGBA值
print(img_arr[0, 0, :])
print(img_tensor[:, 0, 0])

[ 35  33  47 255]
tensor([0.1373, 0.1294, 0.1843, 1.0000])

# 验证, 完全正确
print(255*img_tensor[:, 0, 0])
print(img_arr[0, 0, :])

tensor([ 35.,  33.,  47., 255.])
[ 35  33  47 255]