004pyTorch入门——Transforms
内容目录

合集请看:pyTorch入门合集
参考视频:https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click

transforms在python中如何使用

from PIL import Image
from torchvision import transforms
# 读取图片
img_path = "dataset/train/ants_image/0013035.jpg"
img = Image.open(img_path)
print(img)
# transforms的用法
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
print(tensor_img)

image.png

为什么我们需要Tensor数据类型

Tensor类比python自带的Image类多携带了许多参数,可以理解为是为了方便神经网络计算的一种特殊数据类型

上一篇
下一篇