BBIT-Kai
11 lines
218 B
Python
11 lines
218 B
Python
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
|