1 .pt文件转换为.onxx文件

yolov5s的网络结构文件类型是.pt,但是由于netron不能很好支持.pt可视化,故先把.pt文件转换为.onxx文件,

安装依赖

  1. pip install onnx coremltools onnx-simplifier

导出onxx

  1. python models/export.py --weights yolov5s.pt --img 640 --batch 1
  2. #参数batch? yolov5s代码里有export.py
  3. #这里的pt文件可以跟着目录更改

用Netron打开生成的onxx文件