Pytorch使用OpenCV方式读取单张图片

1. cv2.imread(filename, flags=None)

需要引入头文件: import cv2

参数说明: 

filename: 文件路径，绝对路径和相对路径都可以。

2. 例子

2.1 Code

首先定义路径，然后直接调用函数cv2.imread()即可。

import cv2


def cv2_image():
    image_dir = 'D:/pytorch_code/pytorch_study/fusion_datasets/lytro-01-A.jpg'
    image = cv2.imread(image_dir)
    print("image_shape: ", image.shape)
    print("image_dtype: ", image.dtype)
    print("image_type: ", type(image))
    cv2.imshow('image', image)
    cv2.waitKey(0)


if __name__ == "__main__":
    cv2_image()

2.2 输出显示