"""
    transforms.ToTensor()：将PIL图像 、NP数据转为 Tensor数据类型
"""
from PIL import Image
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter

img_path = "dataset/train/ants/0013035.jpg"
img = Image.open(img_path)

writer = SummaryWriter("logs/004")

# 生成tensor格式的图片
# tensor包装了神经网络全过程使用的一些参数：反向传播、
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)

writer.add_image("Tensor_img", tensor_img)

print(tensor_img)