The issue with image masks being black and white if you try doing multiclass may be due to using PIL L mode when you should instead use P mode. Any chance you’ve figured out how to use the json data directly instead of needing to create the intermediary mask images?