import torchimport torchvision.models as modelsfrom PIL import Imageimport numpy as np#加载测试图片调整大小 image = Image.open("test.jpg") #图片发在了build文件夹下image = image.resize((224, 224),Image.ANTIALIAS)#进行预处理 image = np.asarray(image)image = image / 255image = torch.Tenso...
void Classfier(cv::Mat ){ torch::Tensor img_tensor = torch::from_blob(image.data, {1, image.rows, image.cols, 3}, torch::kByte); img_tensor = img_tensor.permute({0, 3, 1, 2}); img_tensor = img_tensor.toType(torch::kFloat); img_tensor = img_tensor.div(255); std::shared...