基于K类和颜色的彩色图像

时间:2013-07-15 18:33:19

标签: python machine-learning computer-vision

我编写了代码来拍摄图像,分割出重要的对象,并计算有关对象的特征。然后我将对象分解为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]]。然后这些颜色将用于掩盖输入图像,这是我已经可以做的事情,我只需要一种方法来确定颜色。

1 个答案:

答案 0 :(得分:0)

看来你只是想生成K个不同的颜色。这里有很多解决方案: How to automatically generate N "distinct" colors?(它甚至有一个solution in Python,它具有您想要的功能。)