How to start multi-class detection

[zhouhao-learning]

Hello, I am very happy to see your code. Now I have a piece of data. There are 2 types in total. One is benign data, the other is malignant data. They all have corresponding mask map labels. I want to implement multi-classification through Unet. Detecting tasks, but I don't know how to modify the model, and how to process my data for multiple classifications. I don't understand how my label should be defined. Can you teach me more?
Thank you!