pytorch 39 yolov5_obb的onnx部署及其优化

news/2024/7/11 1:56:51 标签: pytorch, YOLO, 深度学习

进行部署要求配置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

#

http://www.niftyadmin.cn/n/245086.html

相关文章

Simulink 自动代码生成电机控制:硬件开发板系统介绍

目录 前言 电源电路 MCU电路 开发板接口 关于电流采样和过流保护 驱动部分 总结 前言 在介绍开发板之前突然有感而发想多说两句,本人从事电控行业也是有一些年头了,除了刚刚毕业就接触的电机控制外,就是电源控制相关的,像三相P…

继续学习Easyx

画圆角矩形函数:roundrect,前四个参数是和矩形一样的,表示先画一个矩形,然后就是要画它的圆角了,要怎么画它的圆角?这里使用四个一样的椭圆来将它的四个角画出来,而在这个里面的椭圆不再需要四个…

OrCAD原理图检查

OrCAD原理图检查 FPGA或处理器芯片原理图封装检查OrCad元件Part Reference与Reference位号不同检查所有器件是否与CIS库元件匹配用CIS库中的元器件替换已存在器件方法1方法2 DRC检查修改页码Annotate重排位号利用Intersheet References功能进行off-page索引检查封装、厂家、型号…

P75分层解耦-IOCDI详解

一、分层解耦-三层架构 Dao层:数据访问 1、接口 package com.itheima.service;import com.itheima.pojo.Emp;import java.util.List;/*** Description:* date: 2023/4/19 21:47** since JDK 11*/ public interface EmpService { // 获取员工列表数据public List&l…

git使用规范文档

git使用规范文档 Git使用规范流程图 开发人员操作步骤: 第一步:clone代码 在你的本地代码库进行从远程仓库clone代码操作(100%表示clone完成) 进入项目文件,右键Git Bash Here 切换到你所进行开发的分支上 拉取该分…

记frp内网穿透配置

这两天由于想给客户看一下我们的系统,于是想到用内网穿透,但是怎么办呢,没有用过呀,于是各处找资料,但是搞完以后已经不记得参考了那些文档了,对不起各位大神,就只能写出过程和要被自己蠢死的错…

AD21原理图----网络连线(网络线、网络标签、总线、差分对、信号线束)

目录 网络连线 网络线(Wire) 网络标签 总线 差分对 信号线束 网络连线 网络线(Wire) 网络标签 可以跨原理图 总线 用于放置同一类数据 使用步骤 第一步:先绘制网络线 第二步:利用网络标签进行连接&a…

ios客户端学习笔记(二):学习Swift基础知识

Swift官方文档 Swift官方文档是学习Swift最权威的资料之一,它包括Swift的语言规范、标准库、编程指南等方面的内容。你可以从Swift官方网站上找到最新的文档:https://swift.org/documentation/ Swift官方文档包括以下内容: 1. The Swift Pr…