PYtorch2 通过cv2读取图片

可以使用cv2.imread()函数来读取图片，然后将其转换为PyTorch张量。以下是一个示例代码：

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()函数将图像的通道维度从(高度，宽度，通道)转换为(通道，高度，宽度)。最后，我们使用torch.from_numpy()函数将numpy数组转换为PyTorch张量。