python - 具有不同形状图像的 Numpy 数组

Question

我需要用不同形状的图像构建一个 numpy 数组来训练一个全卷积网络，该数组需要有 4 级，我需要的形状是 [64, None, None, 3]，以及我的代码使用是这样的：

batch_x = []
for i, j in enumerate(index_array):
  image = cv2.imread(self.filenames[j], is_color)
  image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
  batch_x.append(image)

batch_x = np.array(batch_x)
np.reshape(batch_x, (64, None, None, 3))

但是 numpy 将每个图像视为一个单独的对象。

标签： pythonarraysimagenumpy