如何重塑numpy图像？

Question

我有一个图像作为一个具有形状的numpy数组(channels, height, width)我怎样才能重塑它以使它具有形状(height, width, channels)？

Answer 1

我假设使用内置的numpy函数会更快.

np.rollaxis(array_name,0,3).shape

Answer 2

您可以使用transpose()选择轴的顺序.在这种情况下,你想要:

array.transpose(1, 2, 0)

这样就尽可能在原始数组上创建一个新视图(不会复制任何数据).