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

你可能感兴趣的:(yolo系列改进计划,YOLO)