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