libTorch的部署3包含常用代码解析

    //1.读取文件
    Mat img = imread(“test.jpg”);//opencv读取图片
    //OpenCV读取的都是BGR格式，通常需要转成RGB的
    cv::cvtColor(img , img , cv::COLOR_BGR2RGB);
    
    //2.缩放到模型需要的输入尺寸
    cv::Mat image;
    cv::resize(img, image, cv::Size(256, 256));
    
    //3.将Mat数据转为Tensor张量数据，同时增加了1维
    std::vector<int64_t> sizes = { 1,img.rows, image.cols,3 };
    torch::TensorOptions option(torch::kByte);
    torch::Tensor tensor_image = torch::from_blob(image.data, torch::IntList(sizes), option);
    
    //4.维度换位，来满足模型的输入，就是1X256X256X3 变成 1X3X256X256
    tensor_image = tens