You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
355 B
15 lines
355 B
#coding:utf-8
|
|
from ultralytics import YOLO
|
|
|
|
# 加载模型
|
|
model = YOLO("yolov8n.pt") # 加载预训练模型
|
|
# Use the model
|
|
if __name__ == '__main__':
|
|
# Use the model
|
|
results = model.train(data='datasets/PCB_DATASET/data.yaml', epochs=250, batch=4) # 训练模型
|
|
# 将模型转为onnx格式
|
|
# success = model.export(format='onnx')
|
|
|
|
|
|
|