图像分割输入标签字典到keras

Question

我的训练数据由许多图像组成，每个图像都有一个遮罩。此掩码包含多个类，每个类用单独的颜色表示。你知道吗

如何将每种颜色与标签相关联，以便将其与图像和遮罩一起输入到模型中。你知道吗

我有一本字典，把每种颜色和一个标签联系起来。像这样：

LFW_PARTS_PALETTE = {
    (0, 0, 0) : 0 , 
    (0, 148, 218) : 1 , 
    (90, 90, 90) : 2 , 
    (52,124,49) : 3 , 
    (52,52,255) : 4 , 
    (52,52,137) : 5 , 
    (0,255,255) : 6 , 
    (143,71,143) : 7 , 
    (255,255,255) : 8 , 
    (119,43,43) : 9 , 
    (165,148,44) : 10 , 
}