def generate_color_map()

in a2d2_helpers.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

def generate_color_map(labels_3d):
    """Generates colors to disambiguate classes"""
    label_class_names = sorted(list(set([label["class"] for label in labels_3d.values()])))
    cmap = plt.get_cmap("Set3")
    colors = cmap(np.linspace(0, 1, len(label_class_names)))
    return {label_name: 255*colors[i] for i, label_name in enumerate(label_class_names)}