使用PyTorch和OpenCV读取图片并转换为张量

使用 PyTorch 和 OpenCV (cv2) 读取图片并将其转换为张量是深度学习中常见的操作。以下是如何实现这一过程：

import cv2
import torch

# 读取图片
img = cv2.imread('image.jpg')

# 将图片转换为 PyTorch 张量
img_tensor = torch.from_numpy(img.transpose((2, 0, 1)))

# 打印张量的形状
print(img_tensor.shape)

在这个例子中，我们首先使用 cv2.imread() 函数读取名为 'image.jpg' 的图片。然后，我们使用 NumPy 的 transpose() 函数将图像的通道维度从 (高度，宽度，通道) 转换为 (通道，高度，宽度)，这是 PyTorch 张量所需的格式。最后，我们使用 torch.from_numpy() 函数将 NumPy 数组转换为 PyTorch 张量。