共享博客如何使用 torch.onnx.export 导出的 ONNX 模型进行预测？
调用 torch.onnx.export 导出的 onnx 模型
本文旨在解答如何使用 torch.onnx.export 导出的 onnx 模型。
问题：
如何使用 torch.onnx.export 导出的 onnx.pb 模型文件？
解答：
pytorch 模型的输入为 tensor，而 onnx 的输入为 numpy 数组。因此，将输入数据从 tensor 转换为 numpy 数组即可解决问题。
示例代码：
import onnxruntimeimport numpy as npresnet_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))注意：
文章推荐
-
使用Python获取字符串列表中数字的总和
在Python中，我们可以使用各种方法对字符串列表中的数字进行求和，例如使用正则表达式、使用isdigit()和列表推导...
Python
15秒前 0 -
如何使用TensorFlow Serving在Python中部署模型？
部署机器学习模型对于使人工智能应用程序发挥作用至关重要，为了在生产环境中有效地服务模型，TensorFlow Servi...
Python
53秒前 1 -
Python程序：删除数组/列表中的所有元素的出现次数
数组是存储在连续内存位置的同类数据类型元素的集合。 Python 不提供对内置数组的支持。如果您需要使用数组，则需要导入...
Python
2分钟前 0 -
探秘Python在机器学习中的杰出应用案例
探秘Python在机器学习中的杰出应用案例机器学习已经成为数据科学中不可或缺的一部分。在机器学习的大部分任务中，Pyth...
Python
4分钟前 1 -
Python在智能交通系统中的应用案例分析
Python在智能交通系统中的应用案例分析摘要：随着智能交通系统的快速发展，Python作为一种多功能、易于学习和使用的...
Python
4分钟前 1