博客

在用DANet网络时候，Readme里面提到了处理数据集时，提到了要把cityscapes数据集转换为19类：

• Download the
  
  Cityscapes
  
  dataset and convert the dataset to
  
  19 categories
  
  . It should have this basic structure.

这时如果直接运行会产生下面的报错：

Label Image ID: /home/xxx/data/Cityscapes/gtFine/train/cologne/cologne_000030_000019_gtFine_labelIds.png
Labels can take value between 0 and number of classes.
Some problem with labels. Please check. label_set: [ 0  1  3  4  6  7  8 11 12 17 18 20 21 22 23 24 25 26 33]
…………

不如直接在

https://github.com/erancihan/cityscapesScripts

链接中，下载整个cityscapes的工具包

如图所示，主要用的是

labels.py

和

createTrainIdLabelImgs.py

这两个文件



在

labels.py

文件中，已经给我们改好了需要的19个类的编号。也可以自己根据需要重新改动，255就是忽视这一类的意思，有需要的类可以按照序号编码。这里，文件中已经改好了，所以不赘述。



紧接着，在

createTrainIdLabelImgs.py

文件中，只需要在文件中加上你自己的cityscapes数据集的路径。在这里，我直接将原有的34-37行

注释

掉，然后自己加上了简单的一句

cityscapesPath = '/home/xxx/data/Cityscapes' #cityscapes数据集所在文件夹

如图所示：



然后直接在

createTrainIdLabelImgs.py

文件中右击 run 运行main函数

更改成功！

除此之外也需要对应的

train_list.txt

,

test_list.txt

,

val_list.txt

，如图所示：



所有的文件都放在这里:

https://download.csdn.net/download/qq_41973051/85330186