pytorch中图像的分块（patch）操作（使用了.permute()维度转换）

由于近期代码中有分块的使用，于是进行了搜索总结及探索，如有不正确的，请批评指正。

这里并不能很直观的看到，下面就看一下在图像分块上面的操作

首先给定一张图片

然后在pytorch上读者张图片

from PIL import Image
import torchvision.transforms as transforms
import matplotlib.pyplot as plt


def readImage(path='D:\微信图片_20200304231857.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 = re