查看onnx模型结构-使用Netron模块

查看onnx模型结构-使用Netron模块

1 安装

$pip install netron

2 可选-查看安装的路径

$ pip show netron

3 查看onnx结构

import netron

# �??�?ONNX模�??�??件�??路�?
onnx_model_path = r'yolo5/yolov5n-seg_toXiaoLiu/model/yolov5n-seg.onnx'

# �?�"�netron�?��?��?�??ONNX模�??
netron.start(onnx_model_path)

　　

$python look_onnx.py

4 浏览器中即打开onnx