创建工程,结构如下
创建data.yaml文件
path: C:\Users\1\Desktop\hokai_V1.0.0.202250115.base\xun
train: images/train
val: images/val
nc: 26
names: ['遗器_挑战','指南_生存索引','指南_每日实训','游戏界面','ESC','委托','委托_一键领取','委托_再次派遣','委托_派遣中','指南','每日实训_领取','每日实训_礼物领取','点击空白处关闭','生存索引_侵蚀隧洞','生存索引_侵蚀隧洞_选中','指南_每日实训_选中','指南_生存索引_选中','遗器_迷识之径','遗器_支援','遗器_开始挑战','支援_选人_饮月','支援_选人_饮月_选中','支援_入队','遗器_挑战_再来一次','遗器_挑战_推出关卡','指南_每日实训_满活跃']
手动下载基础模型或者自动下
编写训练模型启动文件
from ultralytics import YOLO
mode = YOLO('yolov8n.pt')
mode.train(
data ='data.yaml',
epochs=500,
imgsz=640,
batch=30,
# device='cpu'
device= 0
)