如何将 4D 图像阵列转换为 3D 图像阵列

Question

image_arr.shape 是 (501, 128, 128, 1)

我的代码：

n_samples, h, w = images_arr.shape

返回以下错误：

ValueError: too many values to unpack (expected 3)

如何将上述形状转换为 3D？

Answer 1

假设数组形状是(501, 128, 128, 1) ，最后一个维度是不需要的，所以我们可以squeeze它挤出来，即。 在这种情况下，所有相关数据都在前 3 个维度中：

images_arr = np.empty((501, 128, 128, 1))

squeezed = np.squeeze(images_arr)

squeezed.shape
>>> (501, 128, 128)

n_samples, h, w = squeezed.shape

Answer 2

(501, 128, 128, 1)是一个 4 元组 - 内部有四个值。 但是， n_samples, h, w需要三个值。

现在请注意，在image_arr.shape中，最后一个值是n_samples, h, w不需要的颜色通道数。

那么你应该做的是：

n_samples, h, w = image_arr.shape[:3]

也就是说，取前三个值（并跳过颜色）。