Yolov5 series (I) -- network visualization tool netron

One 、 Introduce

netron： Network visualization tools , Using this software can clearly show the input and output of each layer of the network model , The overall structure of the network , And support different network frameworks .
Like the image below ：

Two 、 Installation steps （ Introduce windows System installation ）

（ For other versions, please refer to Blog ）

1、 Open this link ：Github Download link

Open the above link , Row down install, choice windows download

2、 Select as shown in the figure below Netron-Setup-5.8.9.exe, Download and install

3、 After successful installation, the following display ：

3、 ... and 、 open yolov5 The model of the series

1、 transformation

because netron Yes .pt Your files are not very compatible , So you need to convert the file , stay yolov5 In Engineering , Provides export.py, File conversion is possible

2、 Directly input the following commands at the terminal

python export.py --weights yolov5s.pt --include onnx

Be careful ： Pay attention when entering commands export.py and yolo5s.pt The path of
I am directly in the directory of this file , So there is no need to write other paths

After executing the order , Will be in yolov5s.pt Generated under the same level directory yolov5s.onnx
At this point, you can use network visualization tools netron open

Four 、 Possible problems in executing commands

requirements: onnx not found and is required by YOLOv5, attempting auto-update...
requirements: 'pip install onnx' skipped (offline)
ONNX: export failure: No module named 'onnx'

terms of settlement

 install onnx
pip install onnx

Then there is no problem to execute the above command