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