plots.py

from matplotlib import cm

def get_pre_define_colors(num_classes, cmap_set: list[str] = None, in_rgb: bool = True, is_float: bool = False):
    if cmap_set is None:
        cmap_set = ['tab20', 'tab20b', 'tab20c']

    colors = []
    for cmap in cmap_set:
        colors.extend(list(cm.get_cmap(cmap).colors))

    if in_rgb:
        new_colors = []
        for color in colors:
            if is_float:
                new_colors.append(tuple(color))
            else:
                new_colors.append(tuple(int(x * 255) for x in color))
        colors = new_colors

    if num_classes > len(colors):
        print(f"WARNING: {num_classes} classes are requested, but only {len(colors)} colors are available. Predefined colors will be reused.")
        colors *= num_classes // len(colors) + 1
    else:
        colors = colors[:num_classes]

    return colors

def xyxy_to_xywh(box):
    return [box[0], box[1], box[2] - box[0], box[3] - box[1]]