> 文章列表 > onnx笔记2:onnx操作实例

onnx笔记2:onnx操作实例

onnx笔记2:onnx操作实例

1. 介绍

本文以yolov5s模型,演示对yolov5s.onnx模型文件的读取,修改等操作
在这里插入图片描述

2. onnx操作

2.1 获取数据

(1) 案例1 :读取weights数据

比如获取yolov5s.onnx第一个Conv的weights数据。
在这里插入图片描述
点击左侧第一个Conv, 右侧INPUTS下面的W点开+号,可以看到该Conv的weight的name为model.0.conv.weight, category为initializer。由于category是initializer,并且name为model.0.conv.weight。因此通过遍历model.graph.initializer就能够很容易定位到weight

import onnxmodel = onnx