YOLOv8-训练自己的数据集

目录

1、上传数据集

2、新建一个traffic.yaml文件

3、修改traffic.yaml文件

4、修改yolov8.yaml文件

5、开始训练

6、验证数据

7、预测数据

---

1、上传数据集

新建一个datasets文件夹，与yolov8同目录下，datasets文件下，存放自己的数据集

数据集如下：

 

2、新建一个traffic.yaml文件

这个位置放在哪里都可以。我是在detect下新建了一个文件夹：mydata，将traffic.yaml放入到mydata文件夹下。

3、修改traffic.yaml文件

train: /root/autodl-tmp/datasets/Traffic/train/images
val: /root/autodl-tmp/datasets/Traffic/valid/images
test: /root/autodl-tmp/datasets/Traffic/test/images
 
nc: 4   
names:
  0: bus
  1: car 
  2: truck
  3: undefined

注意：这里的路径请使用绝对路径，否则会报错。

4、修改yolov8.yaml文件