导入模块
import cv2
import numpy as np
import torch
from torch.autograd import Valiable
格式转换
img = cv2.(img_path)
img = np.resize(img, (32, 32, 3)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # BGR转RGB
img = torch.from_numpy(img.transpose((2, 0, 1))) # transpose维度转换;numpy_to_tensor格式转换
# torch.Size([3, 32, 32])
img = Variable(torch.unsqueeze(img, 0).float, requires_grad=False)
# torch.unsqueeze () 函数主要是对数据维度进行扩充,给指定位置加上维数为1的维度
# Variable函数包装一个Tensor,并记录用在它身上的operations,同时保存着Variable的梯度
处理完成直接放入模型即可
out = model(img)