准备 data.yaml 文件
从roboflow 上下载 CS 游戏数据集, 因为只有CPU, 我对数据集做了瘦身, train: 689张, val: 23张, test:40张.https://universe.roboflow.com/roboflow-100/csgo-videogame/dataset/2
train: ../train/images
val: ../valid/images
test: ../test/images
nc: 2
names: ['CT', 'T']
roboflow:
workspace: roboflow-100
project: csgo-videogame
version: 2
license: CC BY 4.0
url: https://universe.roboflow.com/roboflow-100/csgo-videogame/dataset/2
训练
# 使用 yolov8n.pt 预训练模型进行 train, 不含val
.\yolo task=detect mode=train val=False data=D:\my_workspace\data.yaml model=yolov8n.pt epochs=115 worker