特定のピクセル値の場所を指定した色（RGB）に塗る方法

Question

以下のように特定のピクセル数で構成された画像(image)があります。
構成ピクセル値＝＞「１、２、３、４、５、６」

image.shape # numpy行列
>>>(256, 256, 3)

np.unique(image)
>>>
array([0, 1, 2, 3, 4, 5, 6], type=unit8)

これをimageのピクセル値1と3を黄色や青（などの白黒以外の色）などに塗りたいのですがどうしたら良いのでしょうか？

グレースケールと違い以下の方法ではできませんでした。ご教授お願いします。

img[img==1]=N

Comments

[qax]

構成ピクセル値（０～６）は3チャネルすべて同じ値が格納されているということでしょうか。

[trafalbad]

そうです。セグメンテーション で出力したカラーマップのマスク画像です。

Answer 1

チャネルごとに値を格納する必要があります。

pythonで画像を扱うライブラリはチャネルをRGBで扱うものとBGRで扱うものとがあります。
BGRであるという前提でお話ししますが、基本的には順番を変えるだけですので読み替えてください。

例えば、ピクセル値が1のピクセルを黄色にしたい場合、(b,g,r) = (0, 255, 255)ですから、色(チャネル)ごとに値を書き換える必要があります。

pixel_value = 1
blue_idx = 0
green_idx = 1
red_idx = 2

image[...,blue_idx][image[...,blue_idx] == pixel_value] = 0
image[...,green_idx][image[...,green_idx] == pixel_value] = 255
image[...,red_idx][image[...,red_idx] == pixel_value] = 255

コードの書き方はもっとスマートな書き方があると思いますが、考え方として示しておきます。