Python将图像分割成小块然后将所有的块重新拼接在一起

Posted 2021-11-08 Data+Science+Insight

Python将图像分割成小块然后将所有的块重新拼接在一起

#效果示例

#使用patchify包将图像分割成小块然后将所有的块重新拼接在一起

#二维图像切分及合并

import numpy as np
from patchify import patchify, unpatchify

image = np.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]])

patches = patchify(image, (2,2), step=1) # split image into 2*3 small 2*2 patches.

assert patches.shape == (2, 3, 2, 2)
reconstructed_image = unpatchify(patches, image.shape)

assert (reconstructed_image == image).all()

 #三维图像切分及合并

import numpy as np
from patchify import patchify, unpatchify

image = np.random.rand(512,512,3)

patches = patchify(image, (2,2,3), step=1) # patch shape [2,2,3]
print(patches