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)