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）
|
|||
|
|
)
|