语义分割数据集：Cityscapes的使用

本文主要介绍Cityscapes在语义分割方向上的理解和使用。

Cityscapes官网：官方网站

Cityscapes

简介

Cityscapes大致有两个数据集，分别为精细的标注数据集（3475张训练图像，1525张测试图像）和粗糙的标注数据集（3475+19888张额外的粗糙标注），见图1。

标题

 一般只需要用到精细的部分，也就是4375+1525张图像，在官网直接下载即可，一共5000张。

数据集的原始图片为图2中所示，左边摄像头拍到的图像。共11GB。

图2 原始数据

数据集标注方法

数据集下载以后，需要通过代码文件来生成标注，需要上github下载：cityscapes数据集生成工具在下载好工具后，需要pip安装相应工具包。

pip install cityscapesscripts

在jupyter notebook中也可以输入以下代码进行安装。

!pip install cityscapesscripts

选其一即可。

将下载的工具包打开，进入到preparation文件夹，找到如下文件：打开createTrainIdLabelImages.py

 在其中添加一行代码，保证能读取到你的标注文件路径。

os.environ['CITYSCAPES_DATASET'] = "你的CityScapes  gtFine路径"

 运行createTrainIdLabelImgs.py，即可生成如下数据集（19类）。

生成的数据集-labelTrainIds结尾的图像