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 onnx
model = onnx