ラベル画像のRGB表示の方法について

Question

「keras(tensorflow)でSegNet」参照を参考に
セグメンテーションを試みておりますが正解（ラベル）画像
（教師画像）をRGB表示させる方法が分からず困っております。

CamVid内の下記のような真黒のラベル画像を下記RGB指定したように
RGB表示されるにはどのようにすればよろしいでしょうか。

    Sky = [128,128,128]
    Building = [128,0,0]
    Pole = [192,192,128]
    Road = [128,64,128]   
    Pavement = [60,40,222]
    Tree = [128,128,0]
    SignSymbol = [192,128,128]
    Fence = [64,64,128]
    Car = [64,0,128]
    Pedestrian = [64,64,0]
    Bicyclist = [0,128,192]
    Unlabelled = [0,0,0]

こちらを参考に下記の
コードで一部だけ抽出して表示することはできましたが、全ラベルを
RGB表示できないでおります。

import cv2
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image

pil_img = Image.open("./CamVid/trainannot/0001TP_006690.png")  # PIL で読み込む。
img = np.asarray(pil_img)  # numpy 配列に変換する。

print(img.shape)  # (176, 240)
print(img)


class Label:
    Sky = 0
    Building = 1
    Pole = 2
    Road = 3
    Pavement = 4
    Tree = 5
    SignSymbol = 6
    Fence = 7
    Car = 8
    Pedestrian = 9
    Bicyclist = 10
    Unlabelled = 11

plt.imshow(np.where(img == 3, 100, 0)) # ==　の後の数字を変えるとその領域が表示される Road=3
plt.axis('off')
plt.show()

お手数をお掛けいたしますが、ご教示のほどよろしくお願い申し上げます。

Answer 1

パレットの扱いがよく分かっていませんが、以下のような処理で実現できていそうです。

import matplotlib.pyplot as plt
import numpy as np
from PIL import Image

pil_img = Image.open("inp.png")  # PIL で読み込む。

# カラー番号の確認表示
img = np.asarray(pil_img)  # numpy 配列に変換する。
print(img.shape)  # (176, 240)
print(img)

PAL = [
    [128,128,128], # Sky = 0
    [128,0,0],     # Building = 1
    [192,192,128], # :
    [128,64,128],
    [60,40,222],
    [128,128,0],
    [192,128,128],
    [64,64,128],
    [64,0,128],
    [64,64,0],
    [0,128,192],
    [0,0,0]]
PAL = np.array(PAL).flatten()

# パレットを設定。内部でパレットが作成される？
pil_img.putpalette(PAL)
# パレットを再取得し値を再セット
pal = pil_img.getpalette()
for i in range(len(PAL)):
    pal[i] = PAL[i]
pil_img.putpalette(pal)

plt.imshow(pil_img)
plt.axis('off')
plt.show()

Comments

[hss_]

凄いです！ようやく正解ラベルがRGB表示でみられました。感動です！
本当にありがとうございました。感謝申し上げます。