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

#

猜你喜欢

转载自blog.csdn.net/a486259/article/details/130238663
39