pytorch中transforms的常见使用

from PIL import Image
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter
import os
img_path='3.jpg'
img=Image.open(img_path)
root_dir='data/animaldata/train/ants_image'
img_path_list=os.listdir(root_dir)
#1.totensor的使用
tensor_trans=transforms.ToTensor()
img_tensor=tensor_trans(img)

writer=SummaryWriter('logs')
writer.add_image('totensor',img_tensor)
#1.Normalize的使用
trans_norm=transforms.Normalize([0.5,0.5,0.5],[0.5,0.5,0.5])
img_norm=trans_norm(img_tensor)
writer.add_image('normalize',img_norm)
#3.resize的使用
trans_resize=transforms.Resize((512,512))
img_resize=trans_resize(img)
img_resize_tensor=tensor_trans(img_resize)
writer.add_image('resize',img_resize_tensor)
#4.compose的用法
trans_resize2=transforms.Resize([60,60])
trans_compose=transforms.Compose([trans_resize2,tensor_trans])
img_compose=trans_compose(img)
writer.add_image('compose',img_compose)
#5.RandomCrop的用法
trans_randomcrop=transforms.RandomCrop(80)
trans_compose_2=transforms.Compose([trans_randomcrop,tensor_trans])
for i in range(3):
    train_img_path=os.path.join(root_dir,img_path_list[i])
    train_img=Image.open(train_img_path)
    img_compose_2=trans_compose_2(train_img)
    writer.add_image('RandomCrop',img_compose_2,i)
writer.close()