今天在C++上使用Libtorch的tensor转为Opencv的Mat时总是会出现数据不一致的现象
memcpy(imgMat.data, imgTensor.data_ptr(), imgTensor.numel() * sizeof(torch::kU8));
其中imgMat是我们创建的用来接收Tensor数据的Mat,imgTensor就是结果Tensor。
大家如果输出Mat和Tensor相同位置的数据会发现它们经过memcpy后会有不一致的问题
机缘巧合下也是在以下链接的评论区中
(1条消息) C++ opencv矩阵和pytorch tensor的互相转换_下雨天晴77的博客-CSDN博客
看到这个博主的发言
(1条消息) qq_43632542的博客_CSDN博客-领域博主
我们在memcpy之前对tensor使用一个.contiguous()即可解决这一问题,弄了一下午了。
我只想说,牛!