here's my image &

End result:

Note: If you want fixed color for each mask label, you can create a dictionary and refer from it.