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 输出显示