4.Transform的使用

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

for i in range(50):
    writer.add_scalar("y=i^2", i*i, i)

writer.close()

from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms

writer = SummaryWriter("logs")
img_path = "dataset/train/ants/9715481_b3cb4114ff.jpg"  # 图片路径
img_PIL = Image.open(img_path)                          # 打开图片(PIL)
print(type(img_PIL))    # <class 'PIL.JpegImagePlugin.JpegImageFile'>

trans_tool = transforms.ToTensor()   # 初始化ToTensor对象,不传入参数
tensor_img = trans_tool(img_PIL)     # 传入参数,调用__call__函数(__call__函数需要np或pil类型的参数)
print(tensor_img)

writer.add_image("Tensor_img", tensor_img)  # Tensor类型不用再指定dataformat

writer.close()