使用 MatplotLib 或 Pillow 轉換圖像

Question

我有一張尺寸為 (1000,1000,4) 的圖像，我想將其轉換為 (1000,1000,3)。 如何使用 matplotlib 或 python 中的枕頭模塊執行此操作？

Answer 1

通常，您會將陣列保存為 numpy 陣列，將其展平，然后對其進行整形。 但是，您不能將 (1000,1000,4) 轉換為 (1000,1000,3)。 A (1000,1000,4) 有 1000 x 1000 x 4 個元素，無法轉換為存儲 1000 x 1000 x 3 個元素的數組。 你將不得不擺脫一些元素。

import numpy as np
arr = np.array(your_array) 
arr.flatten().reshape((x dim,y dim ,z dim))