001.png 0 002.png 0 003.png 0 004.png 0 005.png 0中，001.png代表图片名称与类型，0代表“non-broken roofing"分类，怎么把0类变为蓝色，2类变为黑色，1类变为红色，代码

你可以使用Python中的Pillow库来打开和处理图像。以下是一个示例代码，可以将0类转换为蓝色，2类转换为黑色，1类转换为红色。

from PIL import Image

img_path = "001.png"
class_label = 0

# Open the image
img = Image.open(img_path)

# Define the color mapping
color_map = {
    0: (0, 0, 255),  # Blue
    1: (255, 0, 0),  # Red
    2: (0, 0, 0)  # Black
}

# Convert the class label to a color
color = color_map.get(class_label, (0, 0, 0))

# Create a new image with the color
new_img = Image.new('RGB', img.size, color)

# Paste the new image onto the original image
img.paste(new_img, (0, 0), new_img)

# Save the image
img.save("colored.png")

你需要将上面的代码嵌入到你的程序中，循环处理所有的图片，将对应的类别转换成对应的颜色，然后保存成新的图片。