意味分割寸法画像をカラー画像として表示
7867 ワード
意味分割寸法画像をカラー画像として表示
意味分割された寸法画像は、その階調値が分類カテゴリを表すため、記憶されてから直感的に見ることができない(1,2,3,...).寸法画像を直感的に表示するには、次のように対応するカラー画像を生成します.
意味分割された寸法画像は、その階調値が分類カテゴリを表すため、記憶されてから直感的に見ることができない(1,2,3,...).寸法画像を直感的に表示するには、次のように対応するカラー画像を生成します.
def create_visual_anno(anno):
""""""
assert np.max(anno) <= 7, "only 7 classes are supported, add new color in label2color_dict"
label2color_dict = {
0: [0, 0, 0],
1: [255, 248, 220], # cornsilk
2: [100, 149, 237], # cornflowerblue
3: [102, 205, 170], # mediumAquamarine
4: [205, 133, 63], # peru
5: [160, 32, 240], # purple
6: [255, 64, 64], # brown1
7: [139, 69, 19], # Chocolate4
}
# visualize
visual_anno = np.zeros((anno.shape[0], anno.shape[1], 3), dtype=np.uint8)
for i in range(visual_anno.shape[0]): # i for h
for j in range(visual_anno.shape[1]):
color = label2color_dict[anno[i, j]]
visual_anno[i, j, 0] = color[0]
visual_anno[i, j, 1] = color[1]
visual_anno[i, j, 2] = color[2]
return visual_anno