pytorch转onnx 模型输出对不齐

最新推荐文章于 2024-02-03 16:35:16 发布

slamdunkofkd

最新推荐文章于 2024-02-03 16:35:16 发布

阅读量3.6k

点赞数

文章标签： python 人工智能 pytorch

本文链接：https://blog.csdn.net/slamdunkofkd/article/details/112324573

版权

博主在将PyTorch模型转换为ONNX时，遇到28%数据对齐问题，原因在于upsample层只支持nearest模式。通过调整为bilinear并查阅官方文档，解决了问题。建议在迁移模型前检查算子支持情况，尽量使用torch.nn模块。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近，在把pytorch模型转为onnx时，遇到一个问题，有28%的数据对不齐，因为接触的是显著性task的，里面用了一些upsample层。

查了pytorch官方文档后发现，这里的upsample只支持nearest一种模式，而我用的是bilinear，在改变了这个之后，结果就对的齐了。

建议：先去官方文档看一下哪些算子支持哪些算子不支持，以及别用Function函数，得用torch.nn里面的层。

官方文档：https://pytorch.org/docs/stable/onnx.html?highlight=onnx#module-torch.onnx

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

slamdunkofkd

关注关注

0
点赞
踩
3

收藏

觉得还不错? 一键收藏
8
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

模型部署翻车记：pytorch转onnx踩坑实录

Mrrunsen的博客

10-13

4233

在深度学习模型部署时，从pytorch转换onnx的过程中，踩了一些坑。本文总结了这些踩坑记录，希望可以帮助其他人。首先，简单说明一下pytorch转onnx的意义。在pytorch训练出一个深度学习模型后，需要在TensorRT或者openvino部署，这时需要先把Pytorch模型转换到onnx模型之后再做其它转换。因此，在使用pytorch训练深度学习模型完成后，在TensorRT或者openvino或者opencv和onnxruntime部署时，pytorch模型转onnx这一步是必不可少的。接

模型转换 PyTorch转ONNX 入门

热门推荐

黎国溥

02-17

1万+

本文主要介绍如何将PyTorch模型转换为ONNX模型，为后面的模型部署做准备。转换后的xxx.onnx模型，进行加载和测试。最后介绍使用Netron，可视化ONNX模型，看一下网络结构；查看使用了那些算子，以便开发部署。目录前言一、PyTorch模型转ONNX模型1.1 转换为ONNX模型且加载权重1.2 转换为ONNX模型但不加载权重1.3 torch.onnx.export() 函数二、加载ONNX模型三、可视化ONNX模型。

8 条评论您还未登录，请先登录后发表或查看评论

PyTorch-ONNX~精度对齐工具

whaosoft143ai的博客

07-02

563

基于前几篇教程中的知识，本文我们介绍了一个与 PyTorch 转 ONNX 相关的实战项目。其实，这篇教程提到的精度对齐工具还有很大的优化空间，比如：现在测试输入不太方便管理，能不能优化它的管理逻辑？现在我们默认是用 ONNX Runtime 执行 ONNX 模型，可否拓展到其他后端？能不能把无效的 Identity 节点从调试模型中删除？whaosoft aiot http://143ai.com

pth转ONNX模型结果不一致

qq_31638535的博客

01-12

4608

模型转换

pytorch 转换onnx_是什么引起了各个框架 Resize 操作的结果不同？——来自 ONNX 的标准化尝试...

weixin_30781059的博客

12-08

1038

作者：京东AI研究院张建浩炼丹师在转换模型的时候，经常会发现给转换前后的模型输入同样的图片，模型结果有微小的差别。其中的原因有数值算法的误差、不同 jpeg 解码库产生的结果不同等等，也有不同框架内部对某些算子的实现差异。在给 ONNX 贡献 Resize 算子的 spec 的时候，我发现 Resize 是一个突出体现了框架实现差异的算子——多种 Resize 类型、不统一的超参数、将错就错的历...

【RK3588 第四篇】YOLO V5中pytorch2onnx，pytorch和onnx模型输出不一致，精度降低

钱多多先森

10-13

4996

pytorch转onnx模型发现的问题，以及转rknn出现的精度下降，同时验证了新的pytorch转onnx模型出现的麻烦事

pytorch转换onnx,测试onnx,pytorch 模型结果是否一致

m0_37192554的博客

01-07

4866

def to_onnx(): dummy_input = torch.randn(1, 3, 112, 112, dtype=torch.float) # model = model_res() model = model_osnet() input_names = ["data"] output_names = ["fc"] torch.onnx.export( model, dummy_input, "./

pytorch下采样上采样维度无法对齐的问题

Willen_的博客

03-17

6016

pytorch下采样上采样维度无法对齐的问题问题如上，比方说有如下一段pytorch网络代码 model += [nn.Conv2d(ngf * mult, ngf * mult * 2, kernel_size=3, stride=2, padding=1), norm_layer(ngf * mult * 2), activation] .....

将二维数组反转

weixin_39534467的博客

10-23

1067

题目：实现如下转换 $arr = [ [1, 2, 3], [4, 5, 6], [7, 8, 9] ]; 转换后 $arr1 = [ [7, 4, 1], [8, 5, 2], [9, 6, 3] ]; 转换代码如下： function transfer($arr) { $res = []; foreach ($...

Pytorch模型转ONNX模型并运行

最新发布

人工智能

02-03

1014

onnx(Open Neural Network Exchange)主要用于部署，训练一般使用pytorch和tensorflow，等训练完成，我们需要模型转成onnx，用于部署在云或者边缘计算上。而为什么要要转成onnx模型呢，主要是因为onnx没有训练，只有推理，速度很快，而且目前大多数芯片都适配onnx模型，相当于一个通用莫模型，易部署，而且速度快。

pytorch 转onnx模型

weixin_64043217的博客

01-17

755

转模型

pytorch 导出 onnx 模型；onnxruntime在线推理

weixin_42357472的博客

04-11

2185

参考： https://github.com/microsoft/onnxruntime-inference-examples/blob/main/python/OpenVINO_EP/yolov4_object_detection/yolov4.py https://zhuanlan.zhihu.com/p/413992538 ww.cnblogs.com/sddai/p/14537381.html https://www.guyuehome.com/37123 1、pytorch模型保存onnx # c

【ONNX】导出，载入PyTorch的ONNX模型并进行预测新手教程（Windows+Python+Pycharm+PyTorch+ONNX）

AItrust的博客

07-28

1万+

PyTorch是现在非常流行使用的深度学习框架，在我们日常学习过程中用到PyTorch的机会很多。尤其是在科研工作中，由于其使用便利，文档丰富而全面，构建一个实验非常迅速，所以选择使用PyTorch的人也非常多。但是在深度学习模型的应用层面，我们需要将模型在实际使用场景中进行部署，这时候用Python写的模型需要根据需求设置到不同的平台上进行调用，甚至是不同的编程语言去进行调用，这时候模型的转换会是一个非常大的问题。httpshttps//onnx.ai/...

PyTorch语义分割模型转ONNX以及对比转换后的效果（PyTorch2ONNX、Torch2ONNX、pth2onnx、pt2onnx、修改名称、转换、测试、加载ONNX、运行ONNX）

Le0v1n 的博客

08-28

2391

PyTorch2ONNX、Torch2ONNX、pth2onnx、pt2onnx、修改名称、转换、测试、加载ONNX、运行ONNX

pytorch 加载通道不对齐预训练

jacke121的专栏

11-15

2791

pytorch 加载通道不对齐预训练

pytorch 32 onnx模型每次输出的结果都会变的解决方案

a486259的博客

07-21

2706

正常来说onnx模型每次输出的结果都会变，这是一个超乎寻常，难以理解的话题。因为，模型参数一旦导出后，在理论上就不会发生变化。而且，paddle、pytorch、tensorflow等框架导出模型时，也是默认设置模型的mode值为eval，也就是设置参数不更新。如果不是亲身所历，亲眼所示，博主也不会相信有这种匪夷所思的事情。只能说，碰到过的都懂。正常的使用标准ai框架导出部署模型，万不可能发生这种事情。博主之所以碰到这种事情，是因为前期操作秀的太6。把手伸到了MMSegmentation中，，然后。...

可能导致将pytorch转为onnx模型后精度下降的几个原因

m0_46203495的博客

07-02

1万+

可能导致将pytorch转为onnx模型后精度下降的几个原因

Pytorch模型转成onnx并验证结果(排坑经验帖)

木盏

01-14

1万+

很多时候有pytorch模型转onnx模型的必要，比如用tensorRT加速的时候。本文将介绍pytorch的pth模型如何转换成onnx，并且验证你转的模型对不对。先给官网链接：https://pytorch.org/docs/stable/onnx.html 咱们直接用一段代码来看：（本人亲自整理，有问题可留言交流~） import os.path as osp import numpy as np import onnx import onnxruntime as ort import t

pytorch upsample层转onnx报错处理

alex1801

11-25

2227

pytorch转onnx时，上采样层nn.Upsample导出时报错： ONNX's Upsample/Resize operator did not match Pytorch's Interpolation until opset 11. 完整错误： /home/u/miniconda3/envs/torch/lib/python3.7/site-packages/torch/onnx/symbolic_helper.py:267: UserWarning: You are...