【pytorch学习】transforms的使用

ToTensor()

作用：将 PIL Image 类型或者 numpy.ndarray 类型转为 tensor 类型

实例：

from torchvision import transforms
from PIL import Image

img_path = "data/train/bees/17209602_fe5a5a746f.jpg"
img = Image.open(img_path)

print(img)
print(type(img))

tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)

print(tensor_img)
print(type(tensor_img))

运行结果：