from ultralytics import YOLO

def main():
    # 加载你训练好的 YOLO26 模型
    model = YOLO("D:\\ultralytics\\runs\\detect\\train-3\\weights\\best.pt")  # 请将 best.pt 替换为你实际的文件路径

    # 导出为 ONNX 格式
    model.export(
        format="onnx",
        imgsz=(640,384),  # 训练图片大小，默认640
        simplify=True,  # 简化模型结构，提升推理兼容性
        opset=12,  # ONNX 算子版本，适配大多数推理引擎
        dynamic=True  # 支持动态输入尺寸（可选，适合视频等变长输入）
    )
    print("✅ ONNX模型导出成功！")

if __name__ == '__main__':
    main()