yolo训练部署相关代码

2025-11-10 18:08:01 +08:00
parent c098cec233
commit 9527cc2f1c
4 changed files with 286 additions and 0 deletions
@@ -0,0 +1,10 @@
+from ultralytics import YOLO
+import os
+
+# 设置工作目录
+os.chdir("/home/bbit/mine/yolo/")
+
+# 加载模型
+model = YOLO("runs/detect/train13/weights/best.pt")
+
+model.export(format="onnx")  # 默认导出 FP32 ONNX