文章目录 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