Pytorch学习日记——Transforms使用（二）

学习视频——B站【小土堆】

代码

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


# 绝对路径 C:\py\PycharmProjects\pythonproject1\dataset\train\erciyuan_image\407-22641.jpg
# 相对路径 dataset/train/erciyuan_image/407-22641.jpg
img_path ="dataset/train/erciyuan_image/407-22641.jpg"
img = Image.open(img_path)

writer = SummaryWriter("logs")
#print(img)

# transforms该如何使用(python)
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)

writer.add_image("Tensor_img", tensor_img)

writer.close()

在Terminal输入

tensorboard --logdir=logs

 点击蓝色链接

输出：

 以tensor形式显示图片