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


















