class dictionary

[Frozenheart1998]

class dictionary

[DrSleep]

https://github.com/DrSleep/light-weight-refinenet/blob/master/utils/cmap.npy is the colourmap I am using

If you want to see the network's outputs before colouring, simply comment out segm = cmap[segm.argmax(axis=2).astype(np.uint8)] in the notebook

[hyclover]

I recently loaded the cmap.npy through numpy tool. I found that it was an array of form (256,3). However, should it be an array of form (40, 3) since it is the colourmap ? I am curious about what "256" means ?

[DrSleep]

256 just means that there are 256 different colours available.
the NYUD colours only use the first 40 values; VOC -- 20 and etc.

what happens is the network predicts indices from 0 to num_classes, then each predicted index is replaced by a corresponding RGB vector from the cmap file; say, your network predicted indices [2, 3, 1] and the colourmap is [[0, 0, 0], [0, 0, 1], [0, 0, 2], [0, 0, 3]], then the output would be [[0, 0, 2], [0, 0, 3], [0, 0, 1]]

hope that makes it clearer

[Frozenheart1998]

Thank you for your helping!