缩放
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("log1")
trans_resize_1 = transforms.Resize((512, 512))
img_resize = trans_resize_1(img)
tensor_trans_1 = transforms.ToTensor()
img_resize_1 = tensor_trans_1(img_resize)
writer.add_image("Resize1", img_resize_1, 1)
trans_resize_2 = transforms.Resize(512)
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()