pthon依据语义分割标签绘制彩图

要绘制彩色图像，你需要将语义分割标签映射到颜色值。这可以通过创建一个颜色映射表来实现，其中每个类别都被分配了一个唯一的颜色。以下是一个示例代码：

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

# 定义颜色映射表
color_map = {
    0: (0, 0, 0),       # 背景
    1: (128, 0, 0),     # 道路
    2: (0, 128, 0),     # 草地
    3: (128, 128, 0),   # 树木
    4: (0, 0, 128),     # 水体
    5: (128, 0, 128),   # 建筑物
    6: (0, 128, 128),   # 桥梁
    7: (128, 128, 128), # 汽车
}

# 加载语义分割标签
label = np.array(Image.open('label.png'))

# 创建彩色图像
h, w = label.shape
color_label = np.zeros((h, w, 3), dtype=np.uint8)
for i in range(h):
    for j in range(w):
        color_label[i, j, :] = color_map[label[i, j]]

# 显示彩色图像
plt.imshow(color_label)
plt.show()

在这个示例中，我们首先定义了一个颜色映射表，将每个类别分配给一个颜色。然后，我们加载语义分割标签，并将标签映射到相应的颜色值。最后，我们显示彩色图像。