python - 在先 channel 和最后 channel 之间更改图像 channel 顺序的正确方法是什么？

Question

我一生都无法弄清楚如何切换图像排序。图像以 (x,x,3) 格式读取，theano 要求它是 (3,x,x) 格式。我尝试更改顺序numpy.array([img[:,:,i] for i in range(3)])

我想它可以完成工作，但它既丑陋，我不知道如何反转它以恢复原始图像。

Answer 1

我同意@Qualia 的评论，np.moveaxis(a, source, destination)更容易理解。这样就可以了:

x = np.zeros((12, 12, 3))
x.shape
#yields: 
(12, 12, 3)

x = np.moveaxis(x, -1, 0)
x.shape
#yields: 
(3, 12, 12)