search

laugh12321 / TensorRT-YOLO

🚀 你的YOLO部署神器。TensorRT Plugin、CUDA Kernel、CUDA Graphs三管齐下，享受闪电般的推理速度。| Your YOLO Deployment Powerhouse. With the synergy of TensorRT Plugins, CUDA Kernels, and CUDA Graphs, experience lightning-fast inference speeds.
https://github.com/laugh12321/TensorRT-YOLO
GNU General Public License v3.0
720 stars 81 forks source link

[Help]: YOLOv8-OBB 的 ONNX 无法导出 #46

Closed wuaman closed 2 months ago

wuaman commented 2 months ago

通过pip install tensorrt-yolo安装后，导出onnx报错：YOLO version 'yolov8' detect head not supported! yolov8n-obb.pt下载自https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n-obb.pt 望大佬抽空回答，不胜感激 image

laugh12321 commented 2 months ago

@wuaman PiPy 上面的是4.0之前的老版本，新版本需要自己构建才能导出和推理obb，具体的看文档和B站视频

【TensorRT-YOLO】TensorRT 自定义插件加速 YOLO OBB 部署演示-哔哩哔哩】