
机器学习
文章平均质量分 69
亦梦云烟
高性能计算工程师
展开
-
ONNX与JSON相互转换
ONNX模型的修改可以通过转成JSON文件再将JSON文件转回ONNX。以一个Conv算子构成的模型为例:ONNX转JSON使用MessageToJson进行转换,如下将一个conv算子构成的ONNX模型转成JSONimport onnxfrom google.protobuf.json_format import MessageToJson, Parseonnx_model = onnx.load("Conv.onnx")mes...原创 2021-08-18 23:52:32 · 1812 阅读 · 0 评论 -
ONNX构建并运行模型
ONNX是开放式神经网络(Open Neural Network Exchange)的简称,主要由微软和合作伙伴社区创建和维护。很多深度学习训练框架(如Tensorflow, PyTorch, Scikit-learn, MXNet等)的模型都可以导出或转换为标准的ONNX格式,采用ONNX格式作为统一的界面,各种嵌入式平台就可以只需要解析ONNX格式的模型而不用支持多种多样的训练框架,本文主要介绍如何通过代码或JSON文件的形式来构造一个ONNX单算子模型或者整个graph,以及使用ONN...原创 2021-08-07 22:16:23 · 10218 阅读 · 2 评论 -
tflite文件解析
一. 安装flatbuffer下载flatbuffer:https://github.com/google/flatbuffers解压flatbuffer,进入其根目录cd flatbuffers mkdir build cd build cmake .. make -j4 sudo make install flatc --version #输出flatbuffer版...原创 2020-01-19 00:08:43 · 5142 阅读 · 4 评论