import torch
import cv2
from torchvision import transforms
img = torch.from_numpy( cv2.imread('C:/Users/gaojo/Desktop/22.png') ).float()
img = torch.unsqueeze(img,0)
img = img.permute(0,3,1,2)
t=transforms.Compose( [ transforms.Resize( (224, 224) ) ] )
img = t(img)