fix `from_matplotlib` when Matplotlib colormaps does not provide colors property

This commit is contained in:
SkalskiP 2024-07-17 09:42:37 +02:00
parent e76dec7e15
commit 83dd92726e
1 changed files with 10 additions and 4 deletions

View File

@ -352,11 +352,17 @@ class ColorPalette:
supervision-annotator-examples/visualized_color_palette.png)
""" # noqa: E501 // docs
mpl_palette = plt.get_cmap(palette_name, color_count)
colors = [
if hasattr(mpl_palette, 'colors'):
colors = mpl_palette.colors
else:
colors = [mpl_palette(i / (color_count - 1)) for i in range(color_count)]
return cls([
Color(int(r * 255), int(g * 255), int(b * 255))
for r, g, b, _ in mpl_palette.colors
]
return cls(colors)
for r, g, b, _
in colors
])
def by_idx(self, idx: int) -> Color:
"""