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 輸出顯示