无法将输入数组从形状 (2,2,3) 广播到形状 (224,224,3) - 从 NumPy 数组中提取块

Question

我正在尝试通过以下代码从光栅图像（1、4646、4827、3）创建几个具有指定大小的补丁 - 1800、224、224、3：

import numpy as np

image = images_final
tiles = np.zeros((1400, 224, 224, 3))

c = 0

for i in range(0, image.shape[1], 2):  

for j in range(0, image.shape[2], 2):

    tiles[c] = image[:, i:i+2, j:j+2]
    c += 1

但是，我不断收到以下消息：ValueError：无法将输入数组从形状（2,2,3）广播到形状（224,224,3）。

我还尝试了以下代码：

def for_loop_split(image: np.ndarray, kernel_size: tuple):
img_height, img_width, channels = image.shape
tile_height, tile_width = kernel_size

tiled_array = np.zeros ((img_height // tile_height,
                       img_width // tile_width,
                       tile_height,
                       tile_width,
                       channels))
  y = x = 0
  for i in range(0, img_height, tile_height):
  for j in range (0,img_width,tile_width):
  tiled_array[y][x] = image[i:i+tile_height,
                            j:j+tile_width,
                            :channels]
  x += 1
  y += 1
  x = 0 
  return tiled_array

 patches = stride_split(img,tilesize)

但是 output 是 (20, 21, 224, 224, 3)，我需要它是补丁数x 224 x 224 x 3，最后一个对应于光栅的通道。

我知道对于同一个错误有几种解决方案，但它们不适用于我的具体问题。 有人可以帮我吗？