python - How to convert 4D image array to 3D image array

Question

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

My code:

n_samples, h, w = images_arr.shape

Returns the following error:

ValueError: too many values to unpack (expected 3)

How do I convert above shape to 3D ?

标签： pythonarraysnumpy

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