weixin_45415546的博客

如果你想把pytorch训练好的模型快速部署到libtorch上，推荐通过把模型转化为torchscript脚本，然后在libtorch用C++接口加载这个模型。这个方式高效快速把训练模型和工程化快速对接起来。这个方式没有优化前行神经网络计算，但对于一般工程也是够用的,目前在centernet，3d卷积相关网络上成功过。这种的原理大致是通过一次前向传播，把计算图保存下来，后面在python接口或...

libtorch中　torch::from_blob函数　可以把外部内存创建一个tensor，这里给个简单错误示例 torch::Tensor example() { std::vector<float> res_data; res_data.resize(3 * 224* 224); torch::Tensor res_tensor = torch::from_blo...

#include <cuda_runtime_api.h>std::string filename = "centernet.pt"//模型路径int gpu_id = 1; //gpu　id 0代表第一块可见gpucudaSetDevice(gpu_id); //切换显卡torch::jit::script::Module module = torc...

一．获取到网络三个tensor输出hm　(heat map缩写特征图) 得到网络输出后做一个sigmoid()把值映射到0-1wh （width height 回归长宽　这里回归出来长宽是在网络输入长宽）reg（regress 回归中心点偏移　这里回归出来偏移是在网络输入偏移）二．nms 这里和非anchor free方法目前不一样．下面贴出我的c++代码torch::Tensor...