I am working on a project where my task is to identify defects of different types. I use U-net model and performed Binary segmentation and it works. But If I have to perform