一、 transforms的使用

1. 导包

from torchvision import transforms

2. transforms的使用（ToTensor）

  1. from torchvision import transforms
  2. from PIL import Image
  3. # 读取图片
  4. img_path = "data/train/bees/16838648_415acd9e3f.jpg"
  5. img = Image.open(img_path)
  6. # 定义tensorboard
  7. writer = SummaryWriter("logs")
  8. # 定义转换器
  9. tensor_trans = transforms.ToTensor()
  10. # 使用转换器 
  11. img_tensor = tensor_trans(img)
  12. # 展示图片
  13. writer.add_image("TensorImageTest", img_tensor, 1)
  14. writer.close()