Cross-line-detection/test5.py

8 lines
296 B
Python

from ultralytics import YOLO
# 加载模型
model = YOLO('yolov8n.pt') # 假设你使用的是yolov8n.yaml配置文件
# 开始训练,使用自定义参数
model.train(data='C://Users//86195//Desktop//ultralytics-main (1)//ultralytics-main//data.yaml', epochs=5, batch=20, imgsz=640)