大佬们好，我想问下关于windows系统下的yolov5的obb的tensorrt加速有人尝试过吗？

[zkq-afk]

我在使用export文件导出onnx，或者是想生成.wts文件的时候，都遇到了一个报错
AttributeError: Can't get attribute 'OBBModel' on <module 'ultralytics.nn.tasks' from 'C:\Users\DELL\.conda\envs\guangxuanji\lib\site-packages\ultralytics\nn\tasks.py'>，
但是我查找了一下我的文件，发现并没有找到这个OBBModel的类，我想请问一下这个问题怎么解决，或者是关于yolov5的obb的tensorrt加速有其他更好的方法吗？