您现在的位置是：首页 >其他 >pytorch 39 yolov5_obb的onnx部署及其优化网站首页其他

进行部署要求配置opencv和onnxruntime环境，这里不累述。

1、模型导出

yolov5_obb项目的使用可以参考：https://hpg123.blog.csdn.net/article/details/129366477
下载yolov5s_csl_dotav1_best.pt，并执行以下命令，得到yolov5s_csl_dotav1_best.onnx

python export.py --weights ./yolov5s_csl_dotav1_best.pt --device cpu

将导出的模型yolov5s_csl_dotav1_best.onnx上传到https://netron.app/ , 点击输入节点可以发现模型的输入和输出情况如下,yolov5obb是一个多输出模型，其中output节点囊括了所有尺度的输出（这里需要注意output的shape为batchsize,girdnums,grid_pred）。

从中也可以发现模型不支持动态batchsize输入，可以参考 https://hpg123.blog.csdn.net/article/details/130115358?spm=1001.2014.3001.5502 进行修改

2、定义基本函数操作

以下代码保存为Common.hpp

#