open3d.ml.torch.vis.LabelLUT#

class open3d.ml.torch.vis.LabelLUT(label_to_names=None)#

The class to manage look-up table for assigning colors to labels.

class Label(name, value, color)#
__init__(name, value, color)#
__init__(label_to_names=None)#
Parameters:

label_to_names – Initialize the colormap with this mapping from labels (int) to class names (str).

add_label(name, value, color=None)#

Adds a label to the table.

Example

The following sample creates a LUT with 3 labels:

lut = ml3d.vis.LabelLUT()
lut.add_label('one', 1)
lut.add_label('two', 2)
lut.add_label('three', 3, [0,0,1]) # use blue for label 'three'
Parameters:

  • name – The label name as string.

  • value – The value associated with the label.

  • color – Optional RGB color. E.g., [0.2, 0.4, 1.0].

classmethod get_colors(name='default', mode=None)#

Return full list of colors in the lookup table.

Parameters:

  • name (str) – Name of lookup table colormap. Only ‘default’ is supported.

  • mode (str) – Colormap mode. May be None (return as is), ‘lightbg” to move the dark colors earlier in the list or ‘darkbg’ to move them later in the list. This will provide better visual discrimination for the earlier classes.

Returns:

List of colors (R, G, B) in the LUT.

Colors = [[0.0, 0.0, 0.0], [0.96078431, 0.58823529, 0.39215686], [0.96078431, 0.90196078, 0.39215686], [0.58823529, 0.23529412, 0.11764706], [0.70588235, 0.11764706, 0.31372549], [1.0, 0.0, 0.0], [0.11764706, 0.11764706, 1.0], [0.78431373, 0.15686275, 1.0], [0.35294118, 0.11764706, 0.58823529], [1.0, 0.0, 1.0], [1.0, 0.58823529, 1.0], [0.29411765, 0.0, 0.29411765], [0.29411765, 0.0, 0.68627451], [0.0, 0.78431373, 1.0], [0.19607843, 0.47058824, 1.0], [0.0, 0.68627451, 0.0], [0.0, 0.23529412, 0.52941176], [0.31372549, 0.94117647, 0.58823529], [0.58823529, 0.94117647, 1.0], [0.0, 0.0, 1.0], [1.0, 1.0, 0.25], [0.5, 1.0, 0.25], [0.25, 1.0, 0.25], [0.25, 1.0, 0.5], [0.25, 1.0, 1.25], [0.25, 0.5, 1.25], [0.25, 0.25, 1.0], [0.125, 0.125, 0.125], [0.25, 0.25, 0.25], [0.375, 0.375, 0.375], [0.5, 0.5, 0.5], [0.625, 0.625, 0.625], [0.75, 0.75, 0.75], [0.875, 0.875, 0.875]]#