归一化学习演示
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
from PIL import Image
image_path = "./data/train/Cat/222.jpg"
img = Image.open(image_path)
# 创建图表文件
writer = SummaryWriter("logs")
# 数据类型转换
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
# 传入图表数据 原图
writer.add_image("Tensor_img", tensor_img)
# 归一化前
print(tensor_img[0][0][0])
# 零均值归一化 参数:均值、标准差
trans_norm = transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])
img_norm = trans_norm(tensor_img)
# 归一化后的图像
writer.add_image("Normalize", img_norm, 1)
# 归一化后
print(img_norm[0][0][0])
writer.close()
查看命令:
tensorboard --logdir=logs