I’ve trained a 1 class UNet model that can be detect car shapes for segmentation properly, however I need a 10 class classification process with the same dataset. I’m getting good results from only segmentation, so I’m trying to implement this model allows to make classification too.
I labeled my first dataset as I need to, so is there any way to merge this labeled dataset and 1 class segmentation model together and train them properly ?