问题标签 [onnx]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
0 回答
50 浏览

python - 同时将 onnx 模型转移到 tensorflow

ValueError:张量转换为 dtype float64 的张量请求 dtype float32

TypeError:“Add”Op 的输入“y”的类型为 float64,与参数“x”的类型 float32 不匹配

0 投票
0 回答
1093 浏览

keras - ONNX 到 Keras 的转换

我正在使用以下链接进行实施。https://github.com/nerox8664/onnx2keras/

尝试执行代码时出现以下错误。

0 投票
1 回答
663 浏览

tensorflow - tflite 模型到 ONNX 模型的转换

我有一个 tflite 模型,我想将其转换为 tensorflow 或 keras 或 ONNX 格式。有办法吗?

我可以使用 tf 解释器导入它并在 python 上运行它。但是我想将其转换为上述格式之一。

0 投票
1 回答
950 浏览

tensorflow - 如何将 TensorFlow v2 模型转换为 Onnx

我正在尝试将我的 TF 2.0 RC1 模型转换为 onnx(使用 keras2onnx 或 tf2onnx)。当我执行此代码时:

或此代码

我得到错误:

如何解决此错误或以其他方式将我的 Tensorflow v2.0 模型转换为 onnx?

0 投票
2 回答
1105 浏览

c# - 在 C# Windows 窗体应用程序中使用 ONNX 模型

我已经在 Keras 中训练了一个 Mask RCNN 的深度学习模型,并导出了一个能够在 Python 中成功运行和测试图像的 ONNX 模型(权重矩阵)。是否有可能通过 Visual Studio 在 Windows 窗体应用程序、C# 中使用相同的 ONNX 模型?如果是,有什么要求?

系统信息:

  • Windows 10 企业版 2016 LTSB
  • 视觉工作室 2019
  • ONNX 运行时版本(您正在使用):
  • 掩码 RCNN,在 ONNX Runtime v0.5.0 中
0 投票
1 回答
653 浏览

python - 如何在使用 onnx 模型进行预测时解决运行时错误?

我使用trainNetwork命令在 matlab 中训练了深度学习模型。我想在 python 中使用该模型进行预测,所以我在 matlab 中使用“exportONNXNetwork”coomand 将网络导出为 onnx 格式。我使用以下代码在 python 中导入了 onnx 模型: sess = onnxruntime.InferenceSession("Alma.onnx")

该模型接受大小为(224,224,3)的图像。所以我使用 cv2.resize 调整了图像的大小。当我尝试使用sess.run命令运行模型时,我收到一个错误,因为RuntimeError: Input 'data' must not be empty。 其中“数据”是 input_name。用于预测的命令是res = sess.run([output_name], {input_name: x}) 我无法弄清楚我哪里出错了。我正在分享完整的代码。

我得到的错误是:

0 投票
2 回答
358 浏览

machine-learning - 将 Pytorch 图像分类器转换为 mlmodel 时出现问题:无论 img 如何,都返回相同的 softmax 输出

我使用 pytorch 训练和测试了一个图像分类器(Resnet34、Fast.ai、3 类),并按learn.predict()预期工作。当我转换 pytorch -> onnx -> mlmodel 时,无论我提交的图像如何,它都会预测相同的 softmax 值。

这是我的 pytorch 模型:

要将其转换为 .onnx,我需要先对图像数据进行归一化并将其展平。我找到了这个教程,它适用于 fastai/onnx-coreml 的早期版本。我用以下课程做到这一点:

为了构建整个模型,我将 ImageScale 层、模型和 softmax 函数连接起来,如下所示:

最终看起来像这样:

我像这样转换为 .onnx:

我像这样从 .onnx 转换为 .mlmodel :

当我使用 coremltools 调用 predict 时,无论我输入什么图像,我都会得到相同的输出:

可能的问题: 1. 在转换之前我没有正确设置 Sequential 2. Coreml 或 onnx 无法处理非方形图像

我尝试了一堆不同的输入,但一直得到相同的,所以任何帮助将不胜感激!

这是netron的我的头部和尾部的屏幕截图:

头:

头

尾巴:

尾巴

0 投票
1 回答
953 浏览

tensorrt - 编译 tensorrt 样本未找到 NvOnnxParserTypedefs.h

我已经安装了 tensorrt 我已经检查了 python -c "import tensorrt"

但是当我编译 tensorrt 样本时

它抛出

../../include/NvOnnxParser.h:27:10:致命错误:NvOnnxParserTypedefs.h:没有这样的文件或目录#include“NvOnnxParserTypedefs.h”

我在字典中找到包含文件

结果是

谁可以帮助我?也许有一些有用的资源 https://github.com/NVIDIA/TensorRT/blob/release/6.0/samples/opensource/sampleMNIST/README.md https://docs.nvidia.com/deeplearning/sdk /tensorrt-sample-support-guide/index.html#mnist_sample https://docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/index.html#installing-tar https://github.com/NVIDIA /TensorRT/tree/release/5.1 https://github.com/onnx/onnx-tensorrt/tree/5.1 https://github.com/NVIDIA/TensorRT/blob/release/6.0/samples/opensource/sampleMNIST/README .md

0 投票
1 回答
7036 浏览

python - 将 onnx 模型转换为 keras

我尝试将 ONNX 模型转换为 Keras,但是当我调用转换函数时,我收到以下错误消息"TypeError: unhashable type: 'google.protobuf.pyext._message.RepeatedScalarContainer'"

ONNX 模型输入:input_1

您可以在此处查看 ONNX 模型:https ://ibb.co/sKnbxWY

0 投票
1 回答
58 浏览

matlab - 如何在 MATLAB 中导入具有多个输出的模型?

我正在尝试将 ONNX(或 Keras)格式的预训练 Faster R-CNN 模型导入 MATLAB,但是我收到不支持具有多个输出的模型的错误。有没有办法解决这个问题?

我试过 importONNXNetwork 和 importKerasNetwork。

非常感谢您的帮助!