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

Question

我有一个形状[n_samples，color_channels * time_window，宽度，高度] 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)