# 导入ultralytics的YOLO库
from ultralytics import YOLO

# 加载模型
model = YOLO(r'D:\ultralytics-main\ultralytics-main\runs\detect\train5\weights\best.pt')  # 加载自定义的训练模型

# 如果当前脚本作为主程序运行
if __name__ == '__main__':
    model.export(format='onnx')