调用 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))
注意：
将 torch.tensor 转换为 numpy 数组。使用 numpy 数组作为 onnx 模型的输入。onnx 模型的输出是一个列表，包含 numpy 数组。
以上就是如何使用 torch.onnx.export 导出的 ONNX 模型进行预测？的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1352813.html
微信扫一扫 
支付宝扫一扫 
