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