how to export from 'pt' format to 'onnx' format and use Tensor to accelerate detection

iscyy / yoloair

🔥🔥🔥 专注于YOLOv5，YOLOv7、YOLOv8、YOLOv9改进模型，Support to improve backbone, neck, head, loss, IoU, NMS and other modules🚀

https://github.com/iscyy/yoloair

GNU General Public License v3.0

2.47k stars 428 forks source link

Open 4everImmortality opened 10 months ago

4everImmortality commented 10 months ago

ultralytics有 export函数，导出为onnx，yolo-air中没看到这个参数，请问如何把pt模型导出为onnx呢