Pytorch模型部署
文章平均质量分 84
w p h
这个作者很懒,什么都没留下…
展开
-
在ubuntu下将Pytorch模型部署到c++(三)(rknn模型量化)(已完结)
1原创 2021-09-04 14:16:08 · 2509 阅读 · 1 评论 -
在ubuntu下将Pytorch模型部署到c++(二)(模型转换与调用)
模型转换pytorch模型部署到c++需要转化为libtorch可读取的torchscript格式,或者更通用的onnx格式转换torchscript格式net = PoseEstimationWithMobileNet()checkpoint = torch.load("./checkpoint_iter_370000.pth")load_state(net, checkpoint)net.eval()forward_input = torch.rand(1, 3, 256, 394)原创 2021-09-04 13:35:46 · 1101 阅读 · 0 评论 -
在ubuntu下将Pytorch模型部署到c++(一)(环境搭建)
Pytorch模型部署vscode c++环境部署编译器clang安装libtorch安装cudnn安装本文总结pytorch模型部署到c++出现的一些问题,如有错误敬请指正vscode c++环境部署编译器clang安装在ubuntu系统中可直接通过apt命令安装(6.0.0)不过版本较老安装llvmapt-get install llvm2.安装clangapt-get install clang3.检查是否安装成功(查看clang版本,有显示即为成功)clang -v建议在原创 2021-09-04 11:23:47 · 1730 阅读 · 2 评论