如何重塑圖像數據？

Question

我正在嘗試將圖像重塑為我想要的大小。 我在網上有一個圖像數據集，數據集中每張圖像的形狀是(3, 128, 128) 。

我希望每個圖像的形狀為(128, 128, 3) 。 我知道 python 中有幾個函數用於重塑 (np.reshape) 和調整大小 (skimage.transform.resize)，但我不確定這些函數是否適合這種情況。

Answer 1

適合這種情況的 function 是numpy.transpose() 。 假設您有一個形狀為(3, 128, 128)的image ，您可以將其重塑為您想要的尺寸(128, 128, 3)如下所示：

reshaped_image = np.transpose(image, axes=(1, 2, 0))