Tensorflow - deeplab彩色图

Question

我正在处理来自谷歌DeepLab的语义图像分割。我希望能够改变每个语义的颜色(如人，猫等)。使用PASCAL基准测试创建颜色映射的方法是

def create_pascal_label_colormap():
  """Creates a label colormap used in PASCAL VOC segmentation benchmark.

  Returns:
    A Colormap for visualizing segmentation results.
  """
  colormap = np.zeros((256, 3), dtype=int)
  ind = np.arange(256, dtype=int)

  for shift in reversed(range(8)):
    for channel in range(3):
      colormap[:, channel] |= ((ind >> channel) & 1) << shift
    ind >>= 3

  return colormap

我想，如果我用另一个(而不是2 )来改变3的值，我会得到不同的颜色。另外，还有另外一种方法来获取语义的不同颜色吗？--我似乎猜不出它是如何工作的，颜色映射是如何创建的，就像我们在代码中看到的那样，使用移位来创建。我还链接了DeepLab，google：https://colab.research.google.com/drive/1a3TnfeEjVMg7N1Dz5d_UA8GN_iKHkG_l#scrollTo=na9DyxVl4_Ul的完整代码

Answer 1

如果你有固定数量的类，你也可以硬编码你想要的颜色，比如

def create_pascal_label_colormap(): 
 return np.asarray([ 
    [0, 0, 0],
    [0, 192, 0],
    ])