本文主要是介绍yolov5 obb旋转框 tensorrt部署，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
文章目录
- 1.生成engine文件
- 2.检测图像
- 3.代码
yolov5-obb tensorRT部署代码结合王新宇和fish-kong两者的代码，可以多batch批量检测旋转框
yolov5旋转框检测： https://blog.csdn.net/qq_42754919/article/details/134145174
1.生成engine文件
首先需要将pt文件转换成wts文件，执行gen_wts.py文件，修改上面两个文件路径即可。
def parse_args():parser = argparse.ArgumentParser(description这篇关于yolov5 obb旋转框 tensorrt部署的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
