Yolov5系列（一）——網絡可視化工具netron

一、介紹

netron：網絡可視化工具，使用這個軟件可以清晰的展示網絡模型的每一層的輸入輸出，網絡的整體的架構，而且支持不同的網絡框架。
類似下圖：

二、安裝步驟（介紹windows系統安裝）

（其他版本可以參考博客）

1、打開此鏈接：Github下載鏈接

打開上述鏈接，劃到下面 install，選擇 windows 下載

2、選擇下圖所示Netron-Setup-5.8.9.exe，下載安裝

3、安裝成功之後顯示如下：

三、打開yolov5系列的模型

1、轉換

因為netron對 .pt 的文件沒有太好的兼容性，所以需要將文件進行轉換，在yolov5 工程中，提供了export.py，可以進行文件轉換

2、在終端直接輸入以下命令

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

注意：輸入命令時需要注意export.py 和 yolo5s.pt 的路徑
我是直接在該文件的目錄下，所以不需要再寫其他的路徑

執行完命令，會在yolov5s.pt 同級目錄下生成 yolov5s.onnx
此時就可以用網絡可視化工具netron打開

四、執行命令可能遇到的問題

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'

解决辦法

安裝onnx
pip install onnx

之後再執行上述命令就沒有問題了