预测函数:
from ultralytics import YOLO
# Load a model
model = YOLO('yolov8n.pt')
# Train the model
model.train(data=r'D:\yolov8\ultralytics-main\data1.yaml', workers=0, epochs=100, batch=16)
预测函数:
from ultralytics import YOLO
# Load a model
model = YOLO('yolov8n.pt')
# Train the model
model.train(data=r'D:\yolov8\ultralytics-main\data1.yaml', workers=0, epochs=100, batch=16)