将opencv读取的数据转换为tensor

您好，可以使用以下代码将 OpenCV 读取的数据转换为 Tensor：

import torch
import cv2

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

# 转换为 PyTorch Tensor
tensor = torch.from_numpy(img.transpose((2, 0, 1))).float().div(255)

其中，`img.transpose((2, 0, 1))` 将图片的通道维度从最后一维移动到第一维，`float()` 将数据类型转换为浮点型，`div(255)` 将像素值归一化到 [0, 1] 范围内。最终得到的 `tensor` 可以用于 PyTorch 的深度学习模型中。