图像分类ImageNet数据集包含对应多个标签的图像文件夹、标签文件及图像列表文件。 参考数据文件结构如下:
./dataset/ # 数据集根目录
|--labelA # 标签为labelA的图像目录
| |--a1.jpg
| |--...
| └--...
|
|--...
|
|--labelZ # 标签为labelZ的图像目录
| |--z1.jpg
| |--...
| └--...
|
|--train_list.txt # 训练文件列表文件
|
|--val_list.txt # 验证文件列表文件
|
└--labels.txt # 标签列表文件
其中,相应的文件名可根据需要自行定义。
train_list.txt
和val_list.txt
文本以空格为分割符分为两列,第一列为图像文件相对于dataset的相对路径,第二列为图像文件对应的标签id(从0开始)。如下所示:
labelA/a1.jpg 0
labelZ/z1.jpg 25
...
labels.txt
: 每一行为一个单独的类别,相应的行号即为类别对应的id(行号从0开始),如下所示:
labelA
labelB
...
点击这里,下载蔬菜分类分类数据集
在PaddleX中,使用paddlex.cv.datasets.ImageNet
(API说明)加载分类数据集
目标检测VOC数据集包含图像文件夹、标注信息文件夹、标签文件及图像列表文件。 参考数据文件结构如下:
./dataset/ # 数据集根目录
|--JPEGImages # 图像目录
| |--xxx1.jpg
| |--...
| └--...
|
|--Annotations # 标注信息目录
| |--xxx1.xml
| |--...
| └--...
|
|--train_list.txt # 训练文件列表文件
|
|--val_list.txt # 验证文件列表文件
|
└--labels.txt # 标签列表文件
其中,相应的文件名可根据需要自行定义。
train_list.txt
和val_list.txt
文本以空格为分割符分为两列,第一列为图像文件相对于dataset的相对路径,第二列为标注文件相对于dataset的相对路径。如下所示:
JPEGImages/xxx1.jpg Annotations/xxx1.xml
JPEGImages/xxx2.jpg Annotations/xxx2.xml
...
labels.txt
: 每一行为一个单独的类别,相应的行号即为类别对应的id(行号从0开始),如下所示:
labelA
labelB
...
点击这里,下载昆虫检测数据集
在PaddleX中,使用paddlex.cv.datasets.VOCDetection
(API说明)加载目标检测VOC数据集
目标检测和实例分割COCO数据集包含图像文件夹及图像标注信息文件。 参考数据文件结构如下:
./dataset/ # 数据集根目录
|--JPEGImages # 图像目录
| |--xxx1.jpg
| |--...
| └--...
|
|--train.json # 训练相关信息文件
|
└--val.json # 验证相关信息文件
其中,相应的文件名可根据需要自行定义。
train.json
和val.json
存储与标注信息、图像文件相关的信息。如下所示:
{
"annotations": [
{
"iscrowd": 0,
"category_id": 1,
"id": 1,
"area": 33672.0,
"image_id": 1,
"bbox": [232, 32, 138, 244],
"segmentation": [[32, 168, 365, 117, ...]]
},
...
],
"images": [
{
"file_name": "xxx1.jpg",
"height": 512,
"id": 267,
"width": 612
},
...
]
"categories": [
{
"name": "labelA",
"id": 1,
"supercategory": "component"
}
]
}
每个字段的含义如下所示:
域名 | 字段名 | 含义 | 数据类型 | 备注 |
---|---|---|---|---|
annotations | id | 标注信息id | int | 从1开始 |
annotations | iscrowd | 标注框是否为一组对象 | int | 只有0、1两种取值 |
annotations | category_id | 标注框类别id | int | |
annotations | area | 标注框的面积 | float | |
annotations | image_id | 当前标注信息所在图像的id | int | |
annotations | bbox | 标注框坐标 | list | 长度为4,分别代表x,y,w,h |
annotations | segmentation | 标注区域坐标 | list | list中有至少1个list,每个list由每个小区域坐标点的横纵坐标(x,y)组成 |
images | id | 图像id | int | 从1开始 |
images | file_name | 图像文件名 | str | |
images | height | 图像高度 | int | |
images | width | 图像宽度 | int | |
categories | id | 类别id | int | 从1开始 |
categories | name | 类别标签名 | str | |
categories | supercategory | 类别父类的标签名 | str |
点击这里,下载垃圾实例分割数据集
在PaddleX中,使用paddlex.cv.datasets.COCODetection
(API说明)加载COCO格式数据集
语义分割数据集包含原图、标注图及相应的文件列表文件。 参考数据文件结构如下:
./dataset/ # 数据集根目录
|--images # 原图目录
| |--xxx1.png
| |--...
| └--...
|
|--annotations # 标注图目录
| |--xxx1.png
| |--...
| └--...
|
|--train_list.txt # 训练文件列表文件
|
|--val_list.txt # 验证文件列表文件
|
└--labels.txt # 标签列表
其中,相应的文件名可根据需要自行定义。
train_list.txt
和val_list.txt
文本以空格为分割符分为两列,第一列为图像文件相对于dataset的相对路径,第二列为标注图像文件相对于dataset的相对路径。如下所示:
images/xxx1.png annotations/xxx1.png
images/xxx2.png annotations/xxx2.png
...
labels.txt
: 每一行为一个单独的类别,相应的行号即为类别对应的id(行号从0开始),如下所示:
labelA
labelB
...
标注图像为单通道图像,像素值即为对应的类别,像素标注类别需要从0开始递增, 例如0,1,2,3表示有4种类别,标注类别最多为256类。其中可以指定特定的像素值用于表示该值的像素不参与训练和评估(默认为255)。
点击这里,下载视盘语义分割数据集
在PaddleX中,使用paddlex.cv.datasets.SegReader
(API说明)加载语义分割数据集