# export_1280.py
from ultralytics import YOLO

# 加载你的模型
from ultralytics import YOLO

# 加载一个模型，路径为 YOLO 模型的 .pt 文件
model = YOLO(r"/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_obb/weights/best.pt")

# 导出模型，设置多种参数
model.export(
    format="onnx",  # 导出格式为 ONNX
    imgsz=(640, 640),  # 设置输入图像的尺寸
    keras=False,  # 不导出为 Keras 格式
    optimize=False,  # 不进行优化 False, 移动设备优化的参数，用于在导出为TorchScript 格式时进行模型优化
    half=False,  # 不启用 FP16 量化
    int8=False,  # 不启用 INT8 量化
    dynamic=False,  # 不启用动态输入尺寸
    simplify=True,  # 简化 ONNX 模型
    opset=None,  # 使用最新的 opset 版本
    workspace=4.0,  # 为 TensorRT 优化设置最大工作区大小（GiB）
    nms=False,  # 不添加 NMS（非极大值抑制）
    batch=1,  # 指定批处理大小
    device="cpu"  # 指定导出设备为CPU或GPU，对应参数为"cpu" , "0"
)