yolov5s模型转caffe

1.yolov5s模型下载
官网Releases · ultralytics/yolov5 · GitHub 每个版本对应的Assets下有相应的下载链接

2.onnx模型导出

目前我使用的源码为2023年4月6日的master版本,应该对应v7.1或者v8.0版本

主要调用export.py函数进行onnx模型导出,具体执行命令如下

python export.py --weights ../models/*.pt --img 640 --batch 1 --include "onnx"

其中设置图像的分辨率为640*640,batch为1

3. onnx模型转caffe模型

你可能感兴趣的:(YOLO,深度学习,人工智能)