如何重塑存储彩色图像（RGB）序列的numpy数组的数据？

Question

我在形状[n_samples，color_channels * time_window，width，height]中有一个numpy数组存储七帧彩色图像的数据，我想将其重新整形为[n_samples，color_channels，time_window，width，height]如何我能做到吗？

Answer 1

首先：形状是元组。您可以尝试，假设A是您的原始数组：

B = np.reshape(A, (n_samples, color_channels, time_window, width, -1))
print(B.shape)