libtorch Tensor与opencv Mat相互转换

GTX_AI

1. bitmap = bitmap.squeeze().squeeze();
   
2. bitmap = bitmap* 255;
   
3. bitmap = bitmap.to(torch::kU8).to(torch::kCPU);
   
4. int height, width;
   
5. height = bitmap.size(0);
   
6. width = bitmap.size(1);
   
7. cv::Mat imgbin(cv::Size(width, height), CV_8UC1, bitmap.data_ptr());

复制代码

参考：
【1】libtorch Tensor与opencv Mat相互转换
【2】torch::Tensor与opencv mat矩阵转换