YOLOV8多类别训练时遇到的大坑

本篇文章帮大家避坑。。。

如果要训练六个类别的数据集，按照我以下做数据集文件夹

有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