1 .pt文件转换为.onxx文件
yolov5s的网络结构文件类型是.pt,但是由于netron不能很好支持.pt可视化,故先把.pt文件转换为.onxx文件,
安装依赖
pip install onnx coremltools onnx-simplifier
导出onxx
python models/export.py --weights yolov5s.pt --img 640 --batch 1
#参数batch? yolov5s代码里有export.py
#这里的pt文件可以跟着目录更改