【PyTorch笔记】pytorch入门教程4常用的transforms功能

from PIL import  Image
from tensorboardX import SummaryWriter
import torchvision.transforms as transforms
img = Image.open('data/train/ants_image/20935278_9190345f6b.jpg')
writer = SummaryWriter('logs')

#to_tensor
trans_totensor = transforms.ToTensor()
img_tensor = trans_totensor(img)
writer.add_image('ants',img_tensor)


#normalize
print(img_tensor[0][0][0])
trans_norm = transforms.Normalize([0.5,0.5,0.5],[0.5,0.5,0.5])
img_norm = trans_norm(img_tensor)
#__init__(self, mean, std, inplace=False) 因为图片是三维的，所以三个维度的mean和std都要有，就成了列表
print(img_norm[0][0][0])
#img_tensor[0][0][0] 表示 img_tensor 的第一个通道（即红色通道）的第一行第一列的像素值。
writer.add_image('Normalize',img_norm,2)


#resize
print(img.size)
trans_resize = transforms.Resize((512,512))
img_resize = trans_resize(img)
img_resize=trans_totensor(img_resize)
writer.add_image('resize',img_resize,1)
print(img_resize)

#compose
#transforms.Resize((512,512))：将图片的宽度和高度都缩放到 512 像素。如果图片的宽高比与 512:512 不同，那么在缩放过程中会保持宽高比不变，因此可能会出现留白或裁剪的情况。具体来说，如果图片的宽高比大于 512:512，那么在缩放过程中会按照高度进行等比例缩放，直到宽度缩放到 512 像素为止；如果图片的宽高比小于 512:512，那么在缩放过程中会按照宽度进行等比例缩放，直到高度缩放到 512 像素为止。

# transforms.Resize(512)：将图片的短边缩放到 512 像素，长边按照比例进行缩放。具体来说，如果图片的宽度小于高度，那么将图片的宽度缩放到 512 像素，高度按照比例进行缩放；如果图片的高度小于宽度，那么将图片的高度缩放到 512 像素，宽度按照比例进行缩放。这样可以保证图片在缩放过程中不会出现留白或裁剪的情况。
trans_resize_2 = transforms.Resize(512)
trans_compose = transforms.Compose([trans_resize_2,trans_totensor])
img_resize=trans_compose(img)
writer.add_image('resize',img_resize,2)

#randomCrop
trans_random=transforms.RandomCrop((100,300))
trans_compose_2=transforms.Compose([trans_random,trans_totensor])
for i in range (10):
    img_crop=trans_compose_2(img)
    writer.add_image('random',img_crop,i)

writer.close()