簡述

• 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)