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

缩放

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

# 可视化图表使用命令：注意对应名字
# tensorboard --logdir=logs --port=6006
# 可加参数： --port=6007       设置端口号
image_path = "./data/train/Cat/222.jpg"
img = Image.open(image_path)

# 可视化
# 创建图表文件
writer = SummaryWriter("log1")


# 缩放1 缩放至制定大小
trans_resize_1 = transforms.Resize((512, 512))
img_resize = trans_resize_1(img)
# 缩放后图像-->转化为tensor
tensor_trans_1 = transforms.ToTensor()
img_resize_1 = tensor_trans_1(img_resize)
writer.add_image("Resize1", img_resize_1, 1)


# 缩放2 不改变长宽比的缩放
trans_resize_2 = transforms.Resize(512)
# 缩放后图像-->转化为tensor
tensor_trans_2 = transforms.ToTensor()
trans_compose = transforms.Compose([trans_resize_2, tensor_trans_2])
img_resize_2 = trans_compose(img)
writer.add_image("Resize2", img_resize_2, 2)

writer.close()