如何使用 torch.onnx.export 导出的 ONNX 模型进行预测?

ID:21005 / 打印

如何使用 torch.onnx.export 导出的 onnx 模型进行预测?

调用 torch.onnx.export 导出的 onnx 模型

本文旨在解答如何使用 torch.onnx.export 导出的 onnx 模型。

问题:

如何使用 torch.onnx.export 导出的 onnx.pb 模型文件?

解答:

pytorch 模型的输入为 tensor,而 onnx 的输入为 numpy 数组。因此,将输入数据从 tensor 转换为 numpy 数组即可解决问题。

示例代码:

import onnxruntime import numpy as np  resnet_onnx = onnxruntime.InferenceSession('onnx.pb') x = np.ones((2, 2), dtype=np.float32) inputs = {resnet_onnx.get_inputs()[0].name: x} print(resnet_onnx.run(None, inputs))

注意:

  • 将 torch.tensor 转换为 numpy 数组。
  • 使用 numpy 数组作为 onnx 模型的输入。
  • onnx 模型的输出是一个列表,包含 numpy 数组。
上一篇: 使用 Minio SDK 操作阿里云 OSS 可行吗?
下一篇: 如何解决使用 ONNX 模型推理时遇到的 "RuntimeError: Input must be a list of dictionaries or a single numpy array" 错误?

作者:admin @ 24资源网   2025-01-14

本站所有软件、源码、文章均有网友提供,如有侵权联系308410122@qq.com

与本文相关文章

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。