13 lines
399 B
Python
13 lines
399 B
Python
from ultralytics import YOLO
|
||
|
||
# 加载预训练模型
|
||
model = YOLO('../yolov8n.pt')
|
||
|
||
# 开始训练
|
||
model.train(
|
||
data='./yolov8.yaml', # 数据配置文件路径
|
||
epochs=320, # 训练轮数
|
||
batch=-1, # 批量大小
|
||
imgsz=640, # 输入图片大小
|
||
device=0 # 使用的设备（0 表示 GPU，'cpu' 表示 CPU）
|
||
) |