我编写了代码来获取图像、分割重要对象并计算对象的特征。然后我将对象分解为 K 个类,其中 K = 5 - 35 以获得不同的表示。我现在想做的是,给定 K,保存一个蒙版,其中每个对象都是不同的颜色。
我正在努力确定的是一些方法来识别用于不同对象类的颜色。
基本上我正在寻找的是一个功能:
get_rgb(k):
其中 k 是类的数量,返回的是 RGB 列表(或其他颜色表示)。我正在使用 python,我一直在尝试在 matplotlib 中使用 color_maps,但我没有成功。
谢谢你的帮助。
编辑:我正在寻找的是要使用的 K 种不同颜色。所以如果 K = 3,它可能是 [[255,0,0],[0,255,0],[0,0,255]]。然后这些颜色将用于掩盖输入图像,这是我已经可以做的事情,我只需要一种确定颜色的方法。