Open Neural Network Exchange(ONNX,开放神经网络交换)格式,是一个用于表示深度学习模型的标准,可使模型在不同框架之间进行转移(一般用于中间部署阶段)。
ONNX的规范及代码主要由微软,亚马逊 ,Facebook 和 IBM 等公司共同开发,以开放源代码的方式托管在Github上。
目前官方支持加载ONNX模型并进行推理的深度学习框架有: Caffe2, PyTorch, MXNet,ML.NET,TensorRT 和 Microsoft CNTK,并且 TensorFlow 也非官方的支持ONNX
今天弄到了一个ONNX模型,需要查看,记录一下读取查看过程

这是当时搞到的模型,一时间不知道如何下手
后来发现pip有第三方库Netron支持查看,那就直接
pip install netron

这就有了·
然后在.onnx文件所在目录新建一个py文件,编辑加入代码
import netron
netron.start('xuanhuan-2021-10-26.onnx')

会自动弹出浏览器窗口,要是不行就访问http://localhost:8080/

在这个页面等待一下


成功加载
文章评论