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

原创于 2020-11-26 09:02:09 发布 · 599 阅读

1 ·

CC 4.0 BY-SA版权

C/C++ 同时被 3 个专栏收录

12 篇文章

订阅专栏

gcc/g++

6 篇文章

订阅专栏

gdb

3 篇文章

订阅专栏

本文探讨了使用TensorFlow进行图像数据预处理时遇到的内存拷贝问题。通过对比两种不同的数据输入方法，发现直接使用memcpy进行大规模数据复制会导致程序core dump，而采用循环逐个元素复制的方法则能正常运行。文章邀请读者共同讨论问题根源。

//以下写法 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 i=0;i<image.cols * image.rows * 3;i++){
    //input_tensor[i] = image.data[i]/255.0;
    input_tensor[i] = image.data[i];
}

不知道上面一种写法到底错在哪里了, 欢迎跟贴讨论