导出yolov5具有输入输出数据形状（维度）的onnx模型文件

导出yolov5具有输入输出数据形状（维度）的onnx模型文件供netron绘制yolov5模型结构图

参考

1. 导出命令
2. yolov5官方脚本
3. netron 神经网络可视化教程 - 知乎 (zhihu.com)

步骤

1. 修改export.py文件207行，否则无法展现各层输入输出数据的形状（维度）：

    # 增加维度信息
    # onnx.save(model_onnx, f)
    onnx.save(onnx.shape_inference.infer_shapes(model_onnx), f)

2. 执行

python export.py --weights yolov5l6.pt --include onnx --device 0 --opset 11

3. 生成yolov5l6.onnx文件
4. 打开 https://netron.app/ 上传yolov5l6.onnx文件
5. 勾选：
   • Show Attributes：展示各个属性
   • Show weights：展示卷积权重
   • Hide Names：隐藏名字，从而展示处理数据的维度形状（shape）

效果

svg文件