from ultralytics import YOLO

'''
    yolo官方导出的onnx模型转换为rknn模型后并不能使用.
'''
# 加载模型
model = YOLO("/home/admin-root/haotian/rk3588/pytorch模型转rknn/models/yolov8m.pt")

# 导出 ONNX（关键参数！）
model.export(
    format="onnx",
    imgsz=(640, 640),     # 固定输入尺寸
    opset=12,             # ONNX 算子版本 >=12
    simplify=True,        # 简化模型
    dynamic=False,        # 禁用动态轴
    nms=False,            # 移除输出端的 NMS 层（RKNN 不支持）
)

# yolo导出的模型