Pytorch使用PIL的讀取單張圖片並顯示

1. Image.open(fp, mode="r")

調用此方法需要引入頭文件：from PIL import Image。

參數說明：

fp：圖片路徑，可為絕對路徑或相對路徑。

model：默認即可。

2. 例子

2.1 Code

首先給定圖片路徑，然后調用函數Image.open()即可。

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


def plt_image():
    image_dir = 'D:/pytorch_code/pytorch_study/fusion_datasets/lytro-01-A.jpg'
    image = Image.open(image_dir)
    image = np.array(image)
    print("image_shape: ", image.shape)
    print("image_dtype: ", image.dtype)
    print("image_type: ", type(image))
    plt.imshow(image)
    plt.show()


if __name__ == "__main__":
    plt_image()

2.2 輸出顯示