【YOLOv11】014、YOLOv11模型导出:PyTorch到ONNX格式转换与优化

张开发
2026/5/4 2:56:15 15 分钟阅读
【YOLOv11】014、YOLOv11模型导出:PyTorch到ONNX格式转换与优化
从一次深夜调试说起上周团队里有个小伙子跑来找我,说部署好的YOLOv11模型在TensorRT上推理速度比预期慢了近三倍。我让他把转换的ONNX模型发过来一看——好家伙,模型里塞了一堆没融合的算子,动态尺寸没设置对,还带着训练时的冗余节点。这其实是个典型问题:很多人以为torch.onnx.export()一调就完事,结果在部署阶段踩了一堆坑。今天咱们就聊聊怎么把PyTorch训练好的YOLOv11干净利落地转成ONNX,并且让它在实际部署中跑得飞快。转换前准备:模型与数据先看一个常见的错误写法:model=YOLOv11()model.load_state_dict(torch.load('best.pt')

更多文章