画像を取り込み、重要なオブジェクトをセグメント化し、オブジェクトに関する特徴を計算するコードを作成しました。次に、オブジェクトを K = 5 ~ 35 の K クラスに分割して、さまざまな表現を取得しました。私が今やりたいことは、K が与えられ、各オブジェクトが異なる色であるマスクを保存することです。
私が判断するのに苦労しているのは、さまざまなオブジェクト クラスに使用する色を特定する方法です。
基本的に私が探しているのは関数です:
get_rgb(k):
ここで、k はクラスの数であり、戻り値は RGB (またはその他の色表現) のリストです。私はpythonを使用しており、matplotlibでcolor_mapsを使用しようとしていますが、成功していません。
助けてくれてありがとう。
編集:私が探しているのは、使用する K 個の異なる色です。したがって、K = 3 の場合、[[255,0,0],[0,255,0],[0,0,255]] となります。これらの色は、入力画像をマスクするために使用されます。これは既に実行できることですが、色を決定する方法が必要なだけです。