from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
img_path = "D:/05深度学习笔记/Data/FirstTypeData/train/ants/0013035.jpg"
img = Image.open(img_path)
writer = SummaryWriter("logs1")
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
print(tensor_img[0][0][0])
tensor_norm = transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])
norm_img = tensor_norm(tensor_img)
print(norm_img[0][0][0])
writer.add_image("tensor_img", tensor_img)
writer.add_image("norm_img", norm_img)
writer.close()