numpy ndarray和tensor变换

import cv2
import torch

img = cv2.imread("Lena.jpg")
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
print(img)
x = torch.from_numpy(img)
y = x.permute(2, 0, 1)
print(y)