前面我们介绍了旋转目标检测以及如何使用opencv去部署,整体效果不错,但是发现一点就是速度太慢了,所以我们使用TensorRT来部署旋转目标检测算法。感兴趣的同学可以看我之前的关于opencv部署的博客。 https://blog.csdn.net/zhangdaoliang1/article/details/123028071https://blog.csdn.net/zhangdaoliang1/article/details/123028071 目录 1、项目介绍 2、环境配置 3、导出权重文件 4、生成engine 5、模型推理 6、时间对比 7、程序修改的地方 7.1 修改类别数和输入尺寸 7.2 输出类别加1 7.3 yolo层修改 7.4 解码修改 8、代码下载 1、项目介绍 其中的具体效果如下: