专栏名称: 极市平台
极市平台是由深圳极视角推出的专业的视觉算法开发与分发平台,为视觉开发者提供多领域实景训练数据库等开发工具和规模化销售渠道。本公众号将会分享视觉相关的技术资讯,行业动态,在线分享信息,线下活动等。 网站: http://cvmart.net/
目录
相关文章推荐
今天看啥  ›  专栏  ›  极市平台

实操教程|Pytorch转ONNX详解

极市平台  · 公众号  ·  · 2024-07-07 22:00

文章预览

↑ 点击 蓝字  关注极市平台 作者丨立交桥跳水冠军@知乎 来源丨https://zhuanlan.zhihu.com/p/272767300 编辑丨极市平台 极市导读   本文作者总结了自己参与 Pytorch到ONNX的模型转换转换工作中的经验,主要介绍了该转换工作的意义,模型部署的路径以及 Pytorch本身的局限。   >> 加入极市CV技术交流群,走在计算机视觉的最前沿 之前几个月参与了OpenMMlab的模型转ONNX的工作(github account: drcut),主要目标是支持OpenMMLab的一些模型从Pytorch到ONNX的转换。这几个月虽然没做出什么成果,但是踩了很多坑,在这里记录下来,希望可以帮助其他人。 这篇是第一部分,理论篇,主要介绍了和代码无关的一些宏观问题。再接下来我会专门写一篇实战篇,针对OpenMMlab中一些具体代码做分析,说明Pytorch转化ONNX过程中的一些代码上的技巧和注意事项。 (1)Pytorch转ONNX的意义 一 ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览