pytorch 转换onnx_Pytorch转ONNX实战篇2（实战踩坑总结）

最新推荐文章于 2024-06-08 09:49:59 发布
最新推荐文章于 2024-06-08 09:49:59 发布
阅读量1.5k 收藏
点赞数
文章标签： pytorch 转换onnx
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_34620163/article/details/112174111
版权
本文记录了在将Pytorch模型转换为ONNX过程中遇到的问题及解决策略，包括不支持slice赋值、resize操作的限制、kwargs处理、以及Pytorch和ONNX Runtime结果的对齐问题。通过实例展示了如何解决这些问题，确保模型转换的正确性。
摘要由CSDN通过智能技术生成
关注"Smarter"，加" 星标 "置顶

及时获取最优质的CV内容

86d7a3b3e29f38864efa63605b1301a8.png

本文作者：立交桥跳水冠军

https://zhuanlan.zhihu.com/p/286298001

未经授权，严禁二次转载

前两篇文章分别从理论和ONNX的核心机制描述了Pytorch转ONNX需要注意的事情。接下来这篇文章没有什么核心主旨，只是纯粹记录我当时做项目的时候踩的坑以及应对方案

(1)Pytorch2ONNX不支持对slice对象赋值

下面这段代码是不被Pytorch原生的onnx转换接口支持的，即不能对slice对象赋值

preds[:, :, y1:y2, x1:x2] += crop_seg_logit

仔细想想其实也比较合理，因为上面的操作也很难在DAG上被表示，因为并不仅仅是把preds中的那个区域取出来弄个新的变量，然后在上面+1，而是直接把preds的一部分改掉了。当时我负责MMSeg的slide inference转换的

最低0.47元/天 解锁文章
檬chung
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
RuntimeError: Unsupported: ONNX export of Slice with dynamic inputs. DynamicSlice is a deprecated ex
Learning
04-20 4458
问题描述 在使用 torch.onnx.export() 将 Pytorch 版 shufflenet v2 转为 onnx 的时候，出现了一个错误。 出错代码 torch.onnx.export(model, input_example, "{}.onnx".format(modelpath), input_names = None, output_names = None, ) 报错信息 Traceback (most rece
关于pytorch转onnx经常出现的问题
Mr_health的博客
07-25 2039
关于pytorch转onnx经常出现的问题
推荐项目：PyTorch模型轻松转换为TFLite，加速部署之旅
最新发布
gitblog_00100的博客
06-08 379
推荐项目：PyTorch模型轻松转换为TFLite，加速部署之旅 项目地址:https://gitcode.com/sithu31296/PyTorch-ONNX-TFLite 在深度学习的世界里，模型的快速高效部署成为开发者关注的焦点。今天，我们要介绍一个宝藏开源项目——TFLite Conversion，它简化了从PyTorch到TFLite的转换过程，让你的模型部署之路更加顺畅。 1、项目介...
pytorch导出onnx简单易懂（固定输入尺寸以及动态输入尺寸）
m0_49996100的博客
12-08 4379
onnx转pytorch
onnx模型转化总结
PennyYu123的博客
08-30 526
pytorch模型onnx总结
yolov4-tiny通过pytorch导出不带split算子的onnx
轻舟已过万重山的博客
12-08 1380
修改yolov4tiny结构，导出可移植的onnx模型。
onnx_tflite_yolov3:转换工具，可将YOLO v3暗网权重转换为TF Lite模型（YOLO v3 PyTorch> ONNX> TensorFlow> TF Lite）和TensorRT（YOLO v3 Pytorch> ONNX> TensorRT）
05-18
一种转换工具，可将YOLO v3暗网权重转换为TF Lite模型（YOLO v3 PyTorch> ONNX> TensorFlow> TF Lite）和TensorRT模型（dynamic_axes分支）。 先决条件 python3 torch==1.3.1 torchvision==0.4.2 onnx==1.6.0 ...
nnunet-pytorch转onnx
06-13
2. torch.onnx.export 函数：该函数是 PyTorch 提供的将模型转换为 ONNX 模型的 API。该函数需要两个关键参数：model 和 dummy_input。model 是要带有权重的模型，不能光权重参数，dummy_input 是输入参数。 3. ...
TensorRT-使用TensorRT部署Pytorch版YOLOv4-通过Pytorch2onnx导出onnx模型转TRT部署
02-27
总结，本项目实战旨在展示如何利用TensorRT部署PyTorch的YOLOv4模型，通过PyTorch2onnx工具实现模型转换，再通过TensorRT进行性能优化。这一过程不仅提高了模型的推理速度，也简化了跨平台部署的复杂性，对于实际...
pytorch 将onnx模型转换为trt TensorRT模型
03-20
适用于TensorRT8.0版本及以上 使用步骤 1、chmod 添加文件执行权限 2、将onnx路径修改为自己的onnx模型路径 3、运行py问价
nnunet-pytorch转onnx.zip
12-13
nnunet-pytorch转onnx.zip这个压缩包文件包含了将基于PyTorch实现的nnUNet模型转换为ONNX格式的过程，目的是利用ONNX来提高推理速度，并可能进一步运用TensorRT进行加速。nnUNet是一种高度自动化的神经网络框架，专...
Pytorch模型转onnx模型实例
09-18
今天小编就为大家分享一篇Pytorch模型转onnx模型实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
pytorch训练图像分类模型pth转ONNX并测试
10-29
1. 搭建自己的简单二分类网络，使用pytorch训练和测试； 2. 将pytorch训练的pth模型转换成ONNX，并编码测试； 3. 含训练和测试数据，含训练ok的pth模型和ONNX模型，含完整python和C++实现； 4. 使用方法：首先运行“TrainTestConvertOnnx.py”执行“训练数据读入、模型训练、模型测试、导出onnx”，再运行“TestOnnx.cpp”测试onnx（需要配置OpenCV）；
【pytorch】pytorch 转 onnx 踩坑记录
sunnyblogs的博客
09-02 912
import io import torch import torch.onnx import numpy as np from model.detnet import detnet '''onnx的输入为array''' #device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cpu") def test(): model = detnet() model = m
ONNX再探
I good vegetable a!
02-28 1264
0x0. 前言 接着上篇文章，继续探索ONNX。这一节我将主要从盘点ONNX模型部署有哪些常见问题，以及针对这些问题提出一些解决方法，另外本文也会介绍一个可以快速用于ONNX模型推理验证的框架onnxruntime。如果你想用ONNX作为模型转换和部署的工具，可以耐心看下去。今天要讲到的ONNX模型部署碰到的问题大多来自于一些知乎上关于ONNX模型部署的文章以及自己使用ONNX进行模型部署过程中发现的，有一定的实践意义。 0x1. 导出ONNX 这里以Pytorch为例，来介绍一下要把Pytorch模型导
PyTorch学习总结(三)——ONNX
热门推荐
码农王小呆的博客
12-05 5万+
1.什么是ONNXOpen Neural Network Exchange (ONNX)是开放生态系统的第一步，它使人工智能开发人员可以在项目的发展过程中选择合适的工具;ONNX为AI models提供了一种开源格式。它定义了一个可以扩展的计算图模型，同时也定义了内置操作符和标准数据类型。最初我们关注的是推理(评估)所需的能力。Caffe2, PyTorch, Microsoft Cognitive
pytorch2onnx
qq_36113487的博客
09-09 1万+
pytorch2onnx 最近做的项目需要把训练好的模型移植到移动端，安卓手机上，实验室选择了ncnn这个框架，所以我选择了pytoch2onnx2ncnn框架的这个思路。下面主要是记录一下pytorch转onnx模型的步骤和踩过的坑。 项目地址ONNX 定义了一种可扩展的计算图模型、一系列内置的运算单元（OP）和标准数据类型。每一个计算流图都定义为由节点组成的列表，并构建有向无环图。其中每一个节...
pytorch 转换onnx_Pytorch转ONNX详解
weixin_39755952的博客
12-14 1370
点击上方“程序员大白”，选择“星标”公众号重磅干货，第一时间送达作者丨立交桥跳水冠军@知乎来源丨https://zhuanlan.zhihu.com/p/272767300编辑丨极市平台极市导读本文作者总结了自己参与Pytorch到ONNX的模型转换转换工作中的经验，主要介绍了该转换工作的意义，模型部署的路径以及Pytorch本身的局限。之前几个月参与了OpenMMlab的模型转ONN...
模型部署翻车记：pytorch转onnx踩坑实录
nihate的专栏
04-08 4689
本文原创首发于极市平台公众号，如需转载请私信作者 在深度学习模型部署时，从pytorch转换onnx的过程中，踩了一些坑。本文总结了这些踩坑记录，希望可以帮助其他人。 首先，简单说明一下pytorch转onnx的意义。在pytorch训练出一个深度学习模型后，需要在TensorRT或者openvino部署，这时需要先把Pytorch模型转换到onnx模型之后再做其它转换。因此，在使用pytorch训练深度学习模型完成后，在TensorRT或者openvino或者opencv和onnxruntime部署时
如何将pytorch转换成onnx
07-14
2. 导出PyTorch模型：使用PyTorch提供的`torch.onnx.export`函数将PyTorch模型导出为ONNX格式。此函数需要指定输入张量的形状和模型的输入名称。以下是一个示例： ```python import torch # 加载PyTorch模型 ...

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值