原始数据集详情

简介:DLRSD是一个密集的标签数据集,可用于多标签任务,例如遥感图像检索(RSIR)和分类,以及其他基于像素的任务,例如语义分割(在遥感中也称为分类)。

Key Value
卫星类型 GaoFen-1、ZiYuan-3
覆盖区域 未知
场景 未知
分辨率 2m
数量 2100张
单张尺寸 256*256
原始影像位深 8位
标签图片位深 8位
原始影像通道数 三通道
标签图片通道数 单通道

标签类别对照表

像素值 类别名(英文) 类别名(中文) RGB
1 airplane 飞机 166 202 240
2 bare soil 裸土 128 128 0 ]
3 buildings 建筑 0 0 128
4 cars 汽车 255 0 0
5 chaparral 丛林 0 128 0
6 court 庭院 128 0 0
7 dock 码头 255 233 233
8 field 田野 160 160 164
9 grass 草地 0 128 128
10 mobile home 活动房屋 90 87 255
11 pavement 人行道 255 255 0
12 sand 沙子 255 192 0
13 sea 大海 0 0 255
14 ship 255 0 192
15 storage tanks 油罐 128 0 128
16 trees 树林 0 255 0
17 water 水体 0 255 255

数据处理部分

大家是否有这样的困惑,真值标签图片的像素值太小,比如1、2、3······,由于像素值太小,导致看上去标签图片全为黑色,无法看出真值标签与影像图片的像素位置是否对应?如果真值标签的像素值间隔太大,又无法直接作为训练样本。

其实有办法可以解决这个问题,那就是在像素值为1、2、3等的图片上添加一个彩色表,添加的彩色表不会改变图片的像素值,但是可以由彩色的视觉效果展示,如下图所示:

在这里插入图片描述

下面对原始影像及标签进行模型训练前的数据预处理,根据这套数据集的原始信息,原作者整理的已基本接近训练条件,现在需要做的是根据自己训练模型的图片尺寸大小批量裁剪原始影像和标签为所需要的尺寸,比如512*512,另外如果需要检查裁剪后的数据是否一一对应,最好给标签数据添加颜色表,这样就可以轻易看出影像和真值标签是否一致。

下面就是使用代码对原始影像及原始标签进行批量裁剪为512*512大小,并给真值标签添加颜色表,效果图如下所示:

在这里插入图片描述

最后可以直接用于训练的数据集结构如下所示

/path/to/datas
├── train
│   ├── images
│   │   ├── 2522_0_0.tif
│   │   ├── 2522_0_512.tif
│   │   └── ......
│   └── labels
│       ├── 2522_0_0.tif
│       ├── 2522_0_512.tif
│       └── ......
└── val
    ├── images
    │   ├── 2522_512_512.tif
    │   ├── 2523_0_0.tif
    │   └── ......
    └── labels
        ├── 2522_512_512.tif
        ├── 2523_0_0.tif
        └── ......

获取相关数据集或咨询交流见 博主首页 个人简介
1、官网原始数据集;
2、中间处理好的大图数据集;
3、裁剪后可直接训练的小图数据集;

Logo

立足具身智能前沿赛道,致力于搭建全球化、开源化、全栈式技术交流与实践共创平台。

更多推荐