图像分割-输入标签字典以及遮罩到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 , 
}