Categories and Mask Colors

[HolmesJJ]

May I please ask where I can find relationship between the categories and the mask colors? I don't find any keywords of category in the code. I would greatly appreciate it if you could provide me with information.

[Wiill007]

It seems they are all loaded from the checkpoints you're using if you check init_segmentor. What is each class? I don't know at all. However I can tell you each position in checkpoint['meta']['CLASSES'] has a corresponding checkpoint['meta']['PALETTE']. If you print those you can see they are int list for classes and a list of lists of 3 ints for Palette which means basically one color for each class which I guess you already knew. However now at least we know what class number corresponds to each color haha.

Now I'll try to find what is each class wish me luck 😄

[Wiill007]

So I finally got it you see inside the Food103 dataset you'll find category_id.txt which shows all tags used for the checkpoints provided. From what I have seen all checkpoints have been trained for those 103 tags so now you know what tag, number and color corresponds for each class.

[SamyakR99]

Hi @WilsonSmithRobles,

I had a quick follow-up to your previous post,
Where can I find the correspondence between the color palette and the 103 classes in the original dataset (ann_dir folder)?

Thanks!

[Wiill007]

@SamyakR99 I guess it uses a positional reference being color[0] = annotation[0]. I don't know for sure though.

[SamyakR99]

@WilsonSmithRobles

I think that should be it. Thanks for the help!