3.pytorch——torchvision中的transforms

torchvision中的transforms

1.transforms:对图像进行变换的工具库

• tensor：是一种数据类型，我们在使用时经常会将PIL.Image类型的图片转换为tensor类型

代码示例：

from PIL import Image
from torchvision import transforms

#transform的用法 -tensor数据类型
#通过 transform.ToTensor去解决两个问题

#生成PIL.Image类型图片
img_path="data/train/ants_image/0013035.jpg"
img=Image.open(img_path)
print(type(img))

# 1.transform该如何使用（python）
#实例化TOTensor,生成对象对图像进行变换
tensor_trans=transforms.ToTensor()#其实就是生成一个TOTensor工具
tensor_img=tensor_trans(img) #使用该工具将img变为tensor类型，命名为tensor_img

print(tensor_img)

---

• 为什么我们需要Tensor数据类型(如何使用)

  代码示例:

  from PIL import Image
  from torch.utils.tensorboard import SummaryWriter
  from torchvision import transforms
  
  #transform的用法 -tensor数据类型
  #通过 transform.ToTensor去解决两个问题
  # 1.transform该如何使用（python）
  
  
  #生成PIL.Image类型图片
  img_path="data/train/ants_image/0013035.jpg"
  img=Image.open(img_path)
  print(type(img))
  
  #实例化TOTensor,生成对象对图像进行变换
  tensor_trans=transforms.ToTensor()
  tensor_img=tensor_trans(img) #所生成的tensor——img为tensor类型
  
  # 2.为什么我们需要Tensor数据类型(使用Tensor类型数据)
  writer =SummaryWriter("logs")
  writer.add_image("Tensor_img",tensor_img,1)#与pil下的Image类型的使用方法一样
  writer.close()
  
  print(tensor_img)