I am using a U-Net for image segmentaion and Yolo-V3 for localization, they don’t share any layer yet.
I was planning to edit the U-Net by adding just the yolo layer at the end along with the segmentation layers and then train the decoder in the U-Net, but not sure if it’ll work