Transformer的使用
Transform的用途
本文要解决两个问题
- transformers该如何使用(pytorch)?
- 为什么我们需要Tensor数据类型?
transformers该如何使用(pytorch)?
Totensor 的使用方法
from PIL import Image
from torchvision import transforms
#绝对路径 D:\tytorch\Tensorboard\hymenoptera_data\train\ants\0013035.jpg
#相对路径 hymenoptera_data/train/ants/0013035.jpg
img_path="hymenoptera_data/train/ants/0013035.jpg"
img=Image.open(img_path)
tensor_trans=transforms.ToTensor()
#将img转换成tensor
tensor_img=tensor_trans(img)
print(tensor_img)
结果
具体使用的方法:
为什么我们需要Tensor数据类型?
包装了神经网络所需的参数
from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
#绝对路径 D:\tytorch\Tensorboard\hymenoptera_data\train\ants\0013035.jpg
#相对路径 hymenoptera_data/train/ants/0013035.jpg
img_path="hymenoptera_data/train/ants/0013035.jpg"
img=Image.open(img_path)
writer=SummaryWriter("logs")
tensor_trans=transforms.ToTensor()
#将img转换成tensor
tensor_img=tensor_trans(img)
writer.add_image("Tensor_img",tensor_img)
writer.close()
CSDN-Ada助手: 推荐 算法 技能树:https://edu.csdn.net/skill/algorithm?utm_source=AI_act_algorithm
CSDN-Ada助手: 不知道 Python入门 技能树是否可以帮到你:https://edu.csdn.net/skill/python?utm_source=AI_act_python
莫一莫一: 我想问忘记写writer.close()会怎么样?
CSDN-Ada助手: 不知道 Python入门 技能树是否可以帮到你:https://edu.csdn.net/skill/python?utm_source=AI_act_python
简.454: __getitem__就是获取样本对,模型直接通过这一函数获得一对样本对{x:y}。__len__是指数据集长度。