跳转至

YOLOv5 → YOLOv8

简介

将 YOLOv5 格式数据集转换为 YOLOv8 格式。支持标注格式为矩形框和多边形框。

YOLOv5 数据结构如下

Tip

具体结构示例文件，可移步：yolov5_dataset

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
yolov5_dataset
├── classes.txt
├── non_labels  # 通常用来放负样本
│   └── bg1.jpeg
├── images
│   ├── images(13).jpg
│   └── images(3).jpg
├── labels
│   ├── images(13).txt
│   └── images(3).txt
├── train.txt
└── val.txt

转换

1
yolov5_to_yolov8 --data_dir dataset/yolov5_dataset --mode_list train,val
  • --data_dir: 数据集所在目录。示例为 dataset/yolov5_dataset
  • --save_dir: 保存转换后的数据集目录。默认为 dataset/yolov5_dataset_yolov8
  • --mode_list: 指定划分的数据集种类。(例如：train,val,test / train,val)

转换后结构如下

Tip

具体结构示例文件，可移步：yolov8_dataset

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
yolov8_dataset
├── images
│   ├── train
│   │   ├── 0dcddf72.jpg
│   │   └── images(3).jpg
│   └── val
│       ├── 8ae4af51.jpg
│       └── images(13).jpg
└── labels
    ├── train
    │   ├── 0dcddf72.txt
    │   └── images(3).txt
    └── val
        ├── 8ae4af51.txt
        └── images(13).txt

评论