transform函数

1. 用Transform来改变图像大小

from torchvision import transforms
img = cv.imread('./origin/1.jpg')
print('img.shape', img.shape)       # img.shape (426, 640, 3)
# cv.imshow('image', img)
# cv.waitKey(0)

transformer = transforms.Compose([
    transforms.ToPILImage(),
    transforms.Resize((600,800)),		# 600是高，800是宽。
    # transforms.RandomResizedCrop(224,scale=(0.5,1.0)),
    # transforms.RandomHorizontalFlip(),
    transforms.ToTensor(),
    # transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])

img1 = transformer(img)
print('img1.shape', img1.shape)		# img1.shape torch.Size([3, 600, 800])
# cv.imshow("image", img1)
# cv.waitKey(0)