8 lines
296 B
Python
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)
|