创建一个ID分割的图像

时间:2019-08-08 10:39:03

标签: python machine-learning computer-vision dataset image-segmentation

我有一个数据集,其中既包含RGB图像又包含分段图像作为基本事实,包含在带注释的数据集中的readme.txt表示如下:

GT_color:包含用于语义分割的groundtruth掩码的文件夹 注释是使用颜色表示形式给出的,每种颜色都对应于一个特定的类。首先是提供可视化。为了进行训练,请通过将颜色分配给特定的班级ID(如下所示)来创建相应的ID图像。

Class       R   G   B   ID

Void            -   -   -   0

Road            170     170     170 1

Grass           0   255     0   2

Vegetation      102     102     51  3

Tree            0   60  0   3

Sky             0   120     255 4

Obstacle        0   0   0   5

我不明白“创建相应的ID图像”是什么意思,不是每个区域的颜色都已经标记了分段图像吗?意思是rgb是标签?

1 个答案:

答案 0 :(得分:0)

据我了解,您的分割图像带有3个通道,每个像素的颜色与其GT标签相对应。

训练图像分割模型时,不需要输出3个通道(其冗余),因此他们建议您创建一个新的带注释的图像,在其中用提供的i.d替换每种颜色。

此建议只是为了使模型的工作容易一些。