栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

xavier上pytorch pth模型转onnx转tensorrt

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

xavier上pytorch pth模型转onnx转tensorrt

1、pth转onnx

安装onnx从pytorch转换到onnx - 知乎 (zhihu.com)

pip install onnx

遇到问题

AdaptivePooling不能识别,需转为AvgPooling

(307条消息) [开发技巧]·AdaptivePooling与Max/AvgPooling相互转换_简明AI工作室-CSDN博客

(307条消息) torch之AvgPool2d_Nicola.Zhang-CSDN博客_avgpool2d

可视化pytorch模型转换为onnx,并使用netron可视化 - SegmentFault 思否

安装netron

pip install netron
import netron
netron.start('lenet.onnx')
2、onnx转tensorrt

由pytorch模型转onnx及TensorRT - 知乎 (zhihu.com)

(307条消息) TensorRT安装及使用教程_ZONGXP的博客-CSDN博客_tensorrt

TensorRT_C++:加载onnx模型,序列化和反序列化 - 知乎 (zhihu.com)

利用TensorRT实现神经网络提速(读取ONNX模型并运行)_OLDPAN的博客-CSDN博客_tensorrt读取onnx

(1)安装https://www.zybuluo.com/w460461339/note/1679859onnx-to-tensorRThttps://www.zybuluo.com/w460461339/note/1679859

只看文章中的4即可

 链接:https://pan.baidu.com/s/1OlsbvO7eP464oOQWKG10dA?pwd=lx5n 
提取码:lx5n

(2)onnx转为trthttps://blog.csdn.net/weixin_41562691/article/details/118277630

onnx2trt my_model.onnx -o my_engine.trt

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/715484.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号