本篇文章帮大家避坑。。。
如果要训练六个类别的数据集，按照我以下做数据集文件夹
有test、train、valid、还有一个data.yaml的配置文件。
每个下面都有images和labels
我们拿train的imges来说，它里面存放着你六个类别的图片
labels也要跟上面的图片名字一一对应
其次是类别也要对应上。
配置文件怎么写对应怎么写
train: I:/yolov8/ultralytics/ultralytics/datasets/train/images
val: I:/yolov8/ultralytics/ultralytics/datasets/valid/images
test: I:/yolov8/ultralytics/ultralytics/datasets/test/images
nc: 6
names:
0: Missing_hole
1: Mouse_bite
2: Open_circuit
3: Short
4: Spur
5: Spurious_copper然后终端使用命令开始训练！
yolo task=detect mode=train model=./yolov8n.pt data="I:/yolov8/ultralytics/ultralytics/datasets/data.yaml" workers=1 epochs=100 batch=32