![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
lxc_1079
这个作者很懒,什么都没留下…
展开
-
libtorch中数据转换memcpy数据大小匹配
libtorch数据转换经常使用memcpy函数 cv::Mat resultImg(img_h, img_w, CV_8UC3); std::memcpy((void *)resultImg.data, out_tensor.data_ptr(), sizeof(torch::kU8) * out_tensor.numel()); 需要注意的是在c++中float大小是4个字节 但是libtorch中的数据类型并非如此 std::cout << sizeof(torch::kU8) <原创 2020-12-23 15:52:40 · 410 阅读 · 1 评论 -
pytorch 模型有多个返回值 libtorch处理
pytorch 模型有多个返回值 libtorch处理 在python中函数可以有多个返回值,而c++中只允许一个(或者没有)返回值,使用libtorch处理pytorch模型需要对这一点进行处理 1.首先对python中模型进行更改 return F.sigmoid(d0), F.sigmoid(d1), F.sigmoid(d2), F.sigmoid(d3), F.sigmoid(d4), F.sigmoid(d5), F.sigmoid(d6) 需要改为list形式 return [F.sigmo原创 2020-12-21 21:32:51 · 1419 阅读 · 0 评论