深度学习系列4:onnx
发布日期:2021-05-08 09:44:36 浏览次数:18 分类:精选文章

本文共 549 字,大约阅读时间需要 1 分钟。

ONNX 介绍

ONNX(Open Neural Network Exchange)是一种由微软、Facebook 和 AWS 共同开发的深度学习和传统机器学习模型的开放格式。它旨在让算法开发人员能够自由选择机器学习框架,而推理框架的开发人员则可以专注于性能优化和创新。

ONNX Runtime 是基于 ONNX 规范的推理引擎,能够自动调用各种硬件加速器,如 NV CUDA、TensorRT、Intel 的 MKL-DNN 和 nGraph 等。这样,开发者只需将 ONNX 格式的模型传入到蓝色部分的 Runtime,就能完成计算图的分割和并行化处理,只需提供输入数据和输出结果即可。

简单使用指南

以下是一个简单的使用示例:
import onnxruntime as rtsess  
sess = rt.InferenceSession("model.onnx")
input_name = sess.get_inputs()[0].name
X = numpy.random.random((3, 4, 5)).astype(numpy.float32)
pred_onnx = sess.run(None, {input_name: X})
print(pred_onnx)
上一篇:图像处理系列1.skimage
下一篇:PostGis使用总结

发表评论

最新留言

留言是一种美德,欢迎回访!
[***.207.175.100]2025年03月26日 00时38分32秒