Resize裁剪

from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
# 3.3 Resize裁剪
# 3.3.1 Resize裁剪方法一
img_path = "D:/05深度学习笔记/Data/FirstTypeData/train/ants/0013035.jpg"
img = Image.open(img_path)
print(img)  # PIL类型的图片原始比例是768x512

writer = SummaryWriter("logs2")

trans_totensor = transforms.ToTensor()
img_tensor = trans_totensor(img)  # 转化为张量

trans_resize = transforms.Resize((512, 512))
img_resize = trans_resize(img)  # 裁剪

img_resize = trans_totensor(img_resize)  # 裁剪后转化为张量
print(img_resize.size())

writer.add_image("img_tensor", img_tensor)
writer.add_image("img_resize", img_resize)
writer.close()

# Resize裁剪方法二
img_path="D:/05深度学习笔记/Data/FirstTypeData/train/ants/0013035.jpg"
img = Image.open(img_path)

writer = SummaryWriter("logs3")

trans_totensor = transforms.ToTensor()
img_tensor =trans_totensor(img) #转化为张量
print(img_tensor)

#Resize第二种方式：等比缩放
trans_resize_2=transforms.Resize(512) #512/512  512/768
# PIL类型的 Image -> resize -> PIL类型的 Image -> totensor -> tensor类型的 Image
trans_compose=transforms.Compose([trans_resize_2,trans_totensor])
# Compose函数中后面一个参数的输入为前面一个参数的输出
img_resize_2 = trans_compose(img)
print(img_resize_2.size())

writer.add_image("img_tensor", img_tensor)
writer.add_image("img_resize_2", img_resize_2)
writer.close()