python - 将 numpy 图像数组切成 block

Question

我正在使用 python 进行对象检测的图像处理。我需要将我的图像分成所有可能的 block 。例如给定这个玩具图像:

x = np.arange(25)
x = x.reshape((5, 5))

[[ 0  1  2  3  4]
 [ 5  6  7  8  9]
 [10 11 12 13 14]
 [15 16 17 18 19]
 [20 21 22 23 24]]

我想检索给定大小的所有可能 block ，例如 2x2 block 是:

[[0 1]
 [5 6]]
[[1 2]
 [6 7]]

..等等。我该怎么做？

Answer 1

scikit 图片 extract_patches_2d这样做

>>> from sklearn.feature_extraction import image
>>> one_image = np.arange(16).reshape((4, 4))
>>> one_image
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])
>>> patches = image.extract_patches_2d(one_image, (2, 2))
>>> print(patches.shape)
(9, 2, 2)
>>> patches[0]
array([[0, 1],
       [4, 5]])
>>> patches[1]
array([[1, 2],
       [5, 6]])
>>> patches[8]
array([[10, 11],
       [14, 15]])