模型部署(一),onnx,tensorRT,Ncnn学习资料汇总

1、TensorRT-Tensorflow深度学习模型优化视频课程-全套资料分享

https://blog.csdn.net/lqfarmer/article/details/100178743

在这里插入图片描述

1.2 pytorch to onnx to tensorRT

TensorRT5.1.5.0 实践 onnx-TensorRT的自定义op
onnx还需要简化simplify 去除多余的节点信息

不经过caffe 转换, 但是对于自定义层 pytorch to onnx 需要添加源代码 onnx to tensor RT 也需要对应的层

1.1 TensorRT(1)-介绍-使用-安装

https://arleyzhang.github.io/articles/7f4b25ce/

在这里插入图片描述

2 移动端部署ncnn

https://github.com/Tencent/ncnn
https://github.com/BUG1989/caffe-int8-convert-tools.git

在这里插入图片描述
pytorch 得到的模型部署步骤
如果是针对ncnn和tensorRT,那么一般的方案是将pytorch生成的.pth模型先转换成.onnx模型,然后利用onnx的万能属性往不同的框架上移植。
或者转成caffe,再进行下一步操作,都对caffe结构支持较好,都变成caffe ,再转
pytorch2caffe.py,直接转换需要完成。
mxnet2caffe.py,已经完成一次

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值