PEEB / plots.py
Peijie's picture
initial commit
711211a
raw
history blame
948 Bytes
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]]