Python 中將 NumPy 陣列轉換為 PIL 影象

1. 在 Python 中把一個 NumPy 陣列轉換為 PIL 影象
2. 用 Matplotlib Colormap 將 NumPy 陣列轉換為 PIL 影象 Python

本教程解釋了我們如何使用 PIL 包中的 Image.fromarray() 將 NumPy 陣列轉換為 PIL 影象。Python Imaging Library (PIL)是 Python 中一個具有各種影象處理功能的庫。

Image.fromarray() 函式將陣列物件作為輸入，並返回由陣列物件製作的影象物件。

在 Python 中把一個 NumPy 陣列轉換為 PIL 影象

import numpy as np
from PIL import Image

image = Image.open("lena.png")
np_array = np.array(image)

pil_image = Image.fromarray(np_array)
pil_image.show()

輸出：

它將使用 open() 方法從 Image 中讀取當前工作目錄下的影象 lena.png，並返回一個影象物件。

然後我們使用 numpy.array() 方法將這個影象物件轉換為一個 NumPy 陣列。

我們使用 Image.fromarray() 函式將該陣列轉換回 PIL 影象物件，最後使用 show() 方法顯示該影象物件。

import numpy as np
from PIL import Image

array = np.random.randint(255, size=(400, 400), dtype=np.uint8)
image = Image.fromarray(array)
image.show()

輸出：

在這裡，我們建立一個大小為 400x400 的 NumPy 陣列，其隨機數範圍為 0 到 255，然後使用 Image.fromarray() 函式將陣列轉換為 Image 物件，並使用 show() 方法顯示 image。

用 Matplotlib Colormap 將 NumPy 陣列轉換為 PIL 影象 Python

import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
from matplotlib import cm

image_array = plt.imread("lena.jpg")
image_array = image_array / 255
image = Image.fromarray(np.uint8(cm.plasma(image_array) * 255))
image.show()

輸出：

它應用了 Matplotlib 包中的 plasma colormap。要將一個 colormap 應用到一個影象上，我們首先要將 array 歸一化，最大值為 1。在上面的例子中，image_array 中元素的最大值是 255。因此，我們將 image_array 除以 255 進行歸一化。

然後，我們將 colormap 應用到 image_array 中，並再次乘以 255。然後，我們使用 np.uint8() 方法將元素轉換為 int 格式。最後，我們使用 Image.fromarray() 函式將陣列轉換為影象。