16 lines
306 B
Python
16 lines
306 B
Python
# 导入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')
|
|
|
|
|
|
|
|
|
|
|
|
|