对图片进行平均划分

问题描述

pytorch框架中的transforms包含了对图片处理的许多方法，但是没有集成对图片进行平均裁剪的的方法，对此，本文介绍一下参考其他博客的做法。

二、解决方法

代码如下，pytorch框架中图片处理的方式是以四维度的处理的，所以如果处理pytorch中图片时，只需要在mode[ ]内多加一个：即可，就是以mode[:,:,XXX,YYY]方式对图片处理：

from PIL import Image
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
def readImage(path='E:\python_project\测试代码\\1.png', 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)
    image = []
    for ii in range(4):
        for j in range(4):
            if ii == 0 and j == 0:
                image.append(mode[:, 64 * j:64 * (j + 1), 64 * ii:64 * (ii + 1)])
            else:
                image.append(mode[:, 64 * j:64 * (j + 1), 64 * ii:64 * (ii + 1)])
    for i in range(16):
        print(image[i].shape)
        showTorchImage(image[i])

效果示意图

示意图如下所示