About Segmentation Classification Label

[294coder]

Thanks for your open source.
I recently use your code to train for a teeth segmentation, I want to seg a whole teeth(contains gum and teeth) to 4 part separate teeth. Because teeth is just one classification, can I abandon your one-hot label conversion part to train? But now, the IOU is only 80%, can you give me any suggestions?

THANKS!