Netron

Netron对pt格式的权重文件兼容性不好，直接使用netron工具打开，无法显示整个网络。
可使用YOLOV5代码中models/export.py脚本将pt权重文件转换为onnx格式，再用netron工具打开，就可以看YOLOV5网络的整体架构。
pip install onnx>=1.7.0-i https://pypi.tuna.tsinghua.edu.cn/simple # for ONNX export
pip install coremltools==4.0-i https://pypi.tuna.tsinghua.edu.cn/simple # for CoreML export

image.png