前言

我是用yolov5s直接export的onnx，用netron打卡，发现看不到每层的输入，又不想自己算。

步骤

1.准备好onnx

2 执行以下代码

import onnx
onnx_ori="yolov5s.onnx"
onnx_show="yolov5s_show.onnx"
onnx_graph = onnx.load(onnx_ori)
estimated_graph = onnx.shape_inference.infer_shapes(onnx_graph)
onnx.save(estimated_graph, onnx_show)

3.重新使用netron打开