Memcpy在Tensor转为Mat中的小坑

A_RSswjtuer

于 2023-02-24 18:53:29 发布

阅读量306

点赞数 2

分类专栏： Torch 文章标签： opencv 人工智能 c++

本文链接：https://blog.csdn.net/im_nobody33/article/details/129206688

版权

Torch 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

今天在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()即可解决这一问题，弄了一下午了。

我只想说，牛！

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

A_RSswjtuer

关注关注

2
点赞
踩
2

收藏

觉得还不错? 一键收藏
1
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

C++中memcpy和memmove的区别总结

09-04

主要介绍了C++中memcpy和memmove的区别总结,这个问题经常出现在C++的面试题目中,需要的朋友可以参考下

OpenCvSharp 学习笔记22-- 霍夫变换 -像素重映射

C++++ ++

06-16

1359

一，像素重映射介绍：简单的说就是把输入图像中各像素按照一定的规则映射到另一张图像对应的位置上去，形成一张新的图像。 g(x,y)g(x,y)g(x,y) 是重映射之后的图像，h(x,y)h(x,y)h(x,y) 是功能函数，f 是源图像。官方图例：假设我们有一个图像 III : h(x,y)=(I.cols−x,y) h(x,y)=(I.cols-x,y) h(x,y)=(I.cols−...

1 条评论您还未登录，请先登录后发表或查看评论

Mat_memcpy

05-19

图像文件矩阵数据对比工具

[LibTorch] Tensor 与 Mat 相互转换

weixin_43742643的博客

04-30

3544

文章目录单通道三通道CUDA2CPU Torch 框架搭建的神经网络，输入和输出的类型一般是 CUDAFloatType。如果是对图片进行处理，首先使用 OpenCV 读取图片，然后将其转换为 CUDAFloatType，经过神经网络模型进行处理后，再把结果转为 cv::Mat，并通过 imwrite 输出图片。单通道神经网络输出的 tensor 的通道数为 1。 #include <torch/torch.h> #include <torch/script.h> /* 初始

memcpy(input_tensor, image.data, image.total() * image.elemSize()) core dumped

zhe_csdn的博客

11-26

523

//以下写法 core dumped float* input_tensor = interpreter->typed_input_tensor<float>(0); memcpy(input_tensor, image.data, image.total() * image.elemSize()); //修改过的写法正确 float *input_tensor = interpreter->typed_input_tensor<float>(0); for(int

Pytorch 保存tensor到.mat文件

ruotianxia的博客

07-02

2872

Pytorch 保存tensor到.mat文件

CV::Mat数据拷贝 memcpy 踩坑

MAX的专栏

06-27

3706

cv::Mat img1(h, w, CV_8UC3); cv::Mat img1(h, w, CV_8UC3, data);一般cv::mat 初始化后数据内存 data 段都是连续的，可以用isContinuous()函数判断存储是否连续。切记！切记！切记！不连续不能当一维数组处理一般经过裁剪的Mat图像，都不再连续了，如 cv::Mat crop_img = src(rect); crop_img 是不连续的Mat图像，如果想转为连续的，最简单的方法，就是将不连续的crop_img 重新clon

C语言中memcpy 函数的用法详解

08-30

在上面的示例代码中，我们使用memcpy函数将字符串"Pierre de Fermat"拷贝到结构体person的name成员中，然后将结构体person拷贝到结构体person_copy中。最后，我们打印出person_copy的内容。除了使用标准库中的...

一个shape为(3,640,640)的tensor如何转cv::mat

weixin_35749796的博客

01-10

294

一个 3 通道形状为 (3,640,640) 的 Tensor 可以转换成 cv::Mat 类型。你可以使用 cv::Mat 的构造函数来创建一个 cv::Mat，指定其行数、列数和数据类型。然后将 Tensor 的数据拷贝到 cv::Mat 中。例如： cv::Mat mat(640, 640, CV_8UC3); std::memcpy(mat.data, tensor.data_ptr(),...

对比C语言中memccpy()函数和memcpy()函数的用法

12-31

与memcpy()不同的是，memccpy()会在复制时检查参数c 是否出现，若是则返回dest 中值为c 的下一个字节地址。返回值：返回指向dest 中值为c 的下一个字节指针。返回值为0 表示在src 所指内存前n 个字节中没有值为c的...

memcpy（cv::Mat.data,unsigned char *,320x240)，保存数据失真问题解决方法

向阳生长，为梦想而活

05-09

1959

1、memcpy（unsigned char *C，cv::Mat.data A,,320x240)，首先A是图像完整的，通过copy到C中的时候，再copy到新的cv::Mat 图像的对象上时，图像发生失真 memcpy（cv::Mat B.data,unsigned char *C,320x240)，保存数据失真问题解决方法 ...

libtorch常用Mat与Tensor转换

alex1801

11-10

6158

libtorch文档写的真不怎么好，看起来费劲，所以一般用pytoch训练模型，之后把模型保存为traces script model，再通过libtorch加载，进行决策，完成c++集成调用。这里对常用libtorch API进行整理，也只整理常用的。验证环境：win10、vs2017、pytorch1.6、libtorch1.6、cuda10.2、python3.7。 1、Mat转Tensor 1）将图像由BRG转为RGB通道； 2）转为浮点型，量化到0~1； 3）opencv转tenso

c#图像处理（OpenCVSharp）

研究生学习记录

08-27

9057

OpenCVSharp常见图像处理操作（C#）

OpencvSharp 一种读写 float 类型Mat的方法

lvdeyun的博客

12-06

2701

保存 static void SaveMask(Mat mask) { byte[] data = new byte[mask.Width * mask.Height * mask.ElemSize() + MASK_HEADLEN]; BitConverter.GetBytes(mask.Width).CopyTo(data, 0); BitConverter.GetBytes(mask.Height).CopyTo(data, 4); Marshal.Copy(ma.

libtorch 异常处理汇总

一千零一夜的博客

10-20

3362

libtorch 异常处理汇总

c++下memcpy()、memset()、memcmp()需要注意的

weixin_30757793的博客

08-03

2200

memcpy()、memset()、memcmp()等这些内存操作函数经常会帮我们完成一些数据复制、赋值等操作。因为在C语言中，无论是内置类型，还是自定义的结构类型（struct），其内存模型对于我们来说都是可知的、透明的。所以，我们可以对该对象的底层字节序列一一进行操作，简单而有效。代码片段如下所示： structSTUDENT { char_name[32]; ...

Opencv Mat数据类型&操作

yly的博客

10-01

1764

一、概述 Mat用来表示图像或稠密数组。特别注意：数组中的数据不是直接绑定在mat的对象上的。Mat对象实质上是一个数据存储区的头，而非数组本身。如果忽略这点，经常会带来bug。例如，mat1 = mat2，随后修改mat1或mat2中的一个，另一个随之改变，这是因为mat1和mat2两个对象的矩阵头不同，但共享内存空间。上述拷贝属于浅拷贝。深拷贝需要mat1=mat2.clone()。...

c++ malloc memcpy 将多张cv::Mat打包传入python遇到的问题

weihanlai的博客

04-10

677

环境、数据 64位win7 vs2015 c++ 16位灰度图像问题 c++调用Python 来跑神经网络，我需要把将多张cv::Mat打包传入python，四维数组（张数，行，列，通道）我输入了9张图进vector，循环里memcpy完5张到第6张图（即i=5）时程序报错。只输入5张（编号12345）的时候，没有报错，但是输入python的5元数组的图显示出来是：【图1】【全灰】【图...

在C++中，如何将tensor转为多维数组