如何在onehot张量中解码RGB分割？

Question

我想在one-hot分割中转换RGB分割掩码

特别是，给定一个RGB分割掩码y，其中每个像素根据其类别关联到不同的颜色，我想将其转换为一个热分割y_oh，其中每个类别位于不同的通道上

换句话说，y具有形状NxMx3（N=高度，M=宽度，3=通道）y_oh的形状为NxMxC，其中C=类数。最后一个轴是类轴，对于与像素所属类关联的索引，其值为0

我正在研究LIP数据集2，它有19个人体部位的分割标签（例如C=19）。分段贴图如下所示：

谢谢大家!！ G

参考资料：2龚，K，梁，X，沈，X，林，L.：深入人：自我监督结构敏感学习和人类句法分析的新基准。arXiv预印本 arXiv:1703.05446（2017）