用带索引颜色值的tensorflow图像进行语义分割

Question

我正在为语义分割创建FCN。 在PascalVOC数据集中，我很难将标记的png图像转换为索引颜色值。 我希望值在0到20之间。因为我可以在下面的代码中用PIL实现这样的操作

with Image.open(image_path) as img:
    label = np.array(img)

它输出我想要的。但是对于tensorflow实现，我希望它的值与下面的代码相同

^{pr2}$

但是tensorflow实现的结果是0到255之间的值。 有没有什么方法可以在tensorflow中实现PIL实现呢？ 谢谢您。在

Answer 1

SegmentationClass文件中包含颜色映射组件，因此在使用tf.decode_png()时，需要指定为：

label = tf.image.decode_png(file, channels=3)

现在您有了RGB值，您可以使用create_pascal_label_colormap()函数将其转换为类Id