Pytorch模型部署
文章平均质量分 84
w p h
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
在ubuntu下将Pytorch模型部署到c++(三)(rknn模型量化)(已完结)
1原创 2021-09-04 14:16:08 · 2721 阅读 · 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 · 1277 阅读 · 0 评论 -
在ubuntu下将Pytorch模型部署到c++(一)(环境搭建)
Pytorch模型部署vscode c++环境部署编译器clang安装libtorch安装cudnn安装 本文总结pytorch模型部署到c++出现的一些问题,如有错误敬请指正 vscode c++环境部署 编译器clang安装 在ubuntu系统中可直接通过apt命令安装(6.0.0)不过版本较老 安装llvm apt-get install llvm 2.安装clang apt-get install clang 3.检查是否安装成功(查看clang版本,有显示即为成功) clang -v 建议在原创 2021-09-04 11:23:47 · 1932 阅读 · 2 评论
分享