pytorch框架--数据方面--预处理

归一化学习演示

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