| |
|
|
| """ |
| An awesome colormap for really neat visualizations. |
| Copied from Detectron, and removed gray colors. |
| """ |
|
|
| import numpy as np |
| import random |
|
|
| __all__ = ["colormap", "random_color", "random_colors"] |
|
|
| |
| |
| _COLORS = np.array( |
| [ |
| 0.000, 0.447, 0.741, |
| 0.850, 0.325, 0.098, |
| 0.929, 0.694, 0.125, |
| 0.494, 0.184, 0.556, |
| 0.466, 0.674, 0.188, |
| 0.301, 0.745, 0.933, |
| 0.635, 0.078, 0.184, |
| 0.300, 0.300, 0.300, |
| 0.600, 0.600, 0.600, |
| 1.000, 0.000, 0.000, |
| 1.000, 0.500, 0.000, |
| 0.749, 0.749, 0.000, |
| 0.000, 1.000, 0.000, |
| 0.000, 0.000, 1.000, |
| 0.667, 0.000, 1.000, |
| 0.333, 0.333, 0.000, |
| 0.333, 0.667, 0.000, |
| 0.333, 1.000, 0.000, |
| 0.667, 0.333, 0.000, |
| 0.667, 0.667, 0.000, |
| 0.667, 1.000, 0.000, |
| 1.000, 0.333, 0.000, |
| 1.000, 0.667, 0.000, |
| 1.000, 1.000, 0.000, |
| 0.000, 0.333, 0.500, |
| 0.000, 0.667, 0.500, |
| 0.000, 1.000, 0.500, |
| 0.333, 0.000, 0.500, |
| 0.333, 0.333, 0.500, |
| 0.333, 0.667, 0.500, |
| 0.333, 1.000, 0.500, |
| 0.667, 0.000, 0.500, |
| 0.667, 0.333, 0.500, |
| 0.667, 0.667, 0.500, |
| 0.667, 1.000, 0.500, |
| 1.000, 0.000, 0.500, |
| 1.000, 0.333, 0.500, |
| 1.000, 0.667, 0.500, |
| 1.000, 1.000, 0.500, |
| 0.000, 0.333, 1.000, |
| 0.000, 0.667, 1.000, |
| 0.000, 1.000, 1.000, |
| 0.333, 0.000, 1.000, |
| 0.333, 0.333, 1.000, |
| 0.333, 0.667, 1.000, |
| 0.333, 1.000, 1.000, |
| 0.667, 0.000, 1.000, |
| 0.667, 0.333, 1.000, |
| 0.667, 0.667, 1.000, |
| 0.667, 1.000, 1.000, |
| 1.000, 0.000, 1.000, |
| 1.000, 0.333, 1.000, |
| 1.000, 0.667, 1.000, |
| 0.333, 0.000, 0.000, |
| 0.500, 0.000, 0.000, |
| 0.667, 0.000, 0.000, |
| 0.833, 0.000, 0.000, |
| 1.000, 0.000, 0.000, |
| 0.000, 0.167, 0.000, |
| 0.000, 0.333, 0.000, |
| 0.000, 0.500, 0.000, |
| 0.000, 0.667, 0.000, |
| 0.000, 0.833, 0.000, |
| 0.000, 1.000, 0.000, |
| 0.000, 0.000, 0.167, |
| 0.000, 0.000, 0.333, |
| 0.000, 0.000, 0.500, |
| 0.000, 0.000, 0.667, |
| 0.000, 0.000, 0.833, |
| 0.000, 0.000, 1.000, |
| 0.000, 0.000, 0.000, |
| 0.143, 0.143, 0.143, |
| 0.857, 0.857, 0.857, |
| 1.000, 1.000, 1.000 |
| ] |
| ).astype(np.float32).reshape(-1, 3) |
| |
|
|
|
|
| def colormap(rgb=False, maximum=255): |
| """ |
| Args: |
| rgb (bool): whether to return RGB colors or BGR colors. |
| maximum (int): either 255 or 1 |
| |
| Returns: |
| ndarray: a float32 array of Nx3 colors, in range [0, 255] or [0, 1] |
| """ |
| assert maximum in [255, 1], maximum |
| c = _COLORS * maximum |
| if not rgb: |
| c = c[:, ::-1] |
| return c |
|
|
|
|
| def random_color(rgb=False, maximum=255): |
| """ |
| Args: |
| rgb (bool): whether to return RGB colors or BGR colors. |
| maximum (int): either 255 or 1 |
| |
| Returns: |
| ndarray: a vector of 3 numbers |
| """ |
| idx = np.random.randint(0, len(_COLORS)) |
| ret = _COLORS[idx] * maximum |
| if not rgb: |
| ret = ret[::-1] |
| return ret |
|
|
|
|
| def random_colors(N, rgb=False, maximum=255): |
| """ |
| Args: |
| N (int): number of unique colors needed |
| rgb (bool): whether to return RGB colors or BGR colors. |
| maximum (int): either 255 or 1 |
| |
| Returns: |
| ndarray: a list of random_color |
| """ |
| indices = random.sample(range(len(_COLORS)), N) |
| ret = [_COLORS[i] * maximum for i in indices] |
| if not rgb: |
| ret = [x[::-1] for x in ret] |
| return ret |
|
|
|
|
| if __name__ == "__main__": |
| import cv2 |
|
|
| size = 100 |
| H, W = 10, 10 |
| canvas = np.random.rand(H * size, W * size, 3).astype("float32") |
| for h in range(H): |
| for w in range(W): |
| idx = h * W + w |
| if idx >= len(_COLORS): |
| break |
| canvas[h * size : (h + 1) * size, w * size : (w + 1) * size] = _COLORS[idx] |
| cv2.imshow("a", canvas) |
| cv2.waitKey(0) |
|
|
