torch 通过cv2读取图片

以下是使用torch和cv2读取图片的示例代码：

import cv2
import torch

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

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

# 显示图片
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

在上面的代码中，我们首先使用cv2读取图片，然后将其转换为PyTorch张量。最后，我们使用cv2显示图片。请注意，我们需要将图片的通道维度从最后一个维度移动到第一个维度，以便PyTorch能够正确地解释图像。