下载yolov3并转onnx

终端

git clone https://hub.fastgit.org/ultralytics/yolov3.git

cd 进入文件夹yolov3,运行shell脚本下载模型yolov3-tiny.pt

./weights/download_weights.sh

安装依赖库

pip install onnx coremltools onnx-simplifier

导出 onnx

python models/export.py --weights yolov3-tiny.pt --img 640 --batch 1

 

netron查看yolov3-tiny.onnx模型结构。

# 在终端运行
pip install netron
python
import netron
netron.start('yolov3-tiny.onnx')