我已经编写了代码来处理输入图像，分割出重要对象，并计算这些对象的特征。接着，我将这些对象分成了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]]。这些颜色将用于掩盖输入图像，这是我已经能做到的，我只是需要一种确定颜色的方法。

回答：

看起来你只是想生成K种不同的颜色。这里已经有很多解决方案了：如何自动生成N种“不同”的颜色？（甚至有一个Python解决方案，它正好是你想要的函数）。