版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/a19990412/article/details/84793917
简述
- Scale是做比例放缩
- CenterCrop是切割
代码
from PIL import Image
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
def readImage(path='../mode.jpg', size=256):
mode = Image.open(path)
transform1 = transforms.Compose([
transforms.Scale(size),
transforms.CenterCrop((size, size)),
transforms.ToTensor()
])
mode = transform1(mode)
return mode
def showTorchImage(image):
mode = transforms.ToPILImage()(image)
plt.imshow(mode)
plt.show()
if __name__ == '__main__':
mode = readImage(size=256)
showTorchImage(mode)