將圖像尺寸 (a, z, y, x) 更改為 (x, y, z)

Question

我有一個 NumPy 數組，它表示形狀為 (1, 3, 480, 640) 的圖像。 我如何把它變成一個形狀（640、480、3）？

謝謝

Answer 1

您可以使用numpy.squeeze和numpy.swapaxes 。

img = np.random.rand(1,3,480,640)
img = np.squeeze(img)
print(img.shape)
# (3, 480, 640)

img = np.swapaxes(img, axis1=2, axis2=0)
print(img.shape)
# (3, 480, 640)