pytorch讀取圖片並按比例改變圖片的大小或者是固定大小
阿新 • • 發佈:2018-12-12
簡述
- 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)