Segmentation and keypoints training

[salvaba94]

Feature Description

I know there are traces of the repo being ready for instance segmentation inference but not for training. I probably won't be speaking only for myself but also for the rest following this amazing repo. Consider adding instance/panoptic segmentation like the original repo had or going beyond that with keypoints. I think it will add a lot of value to this MIT-licensed repo.

[SofianeCordon]

Hi,
I probably encountered the same issue trying to train a segmentation model using this repo (YOLO-v9-c-seg). Have you found any solution?
Thanks.

[salvaba94]

Segmentation training is not implemented, there is no segmentation loss if you look for it, dataloader does not support segmentation, neither do the augmentations. Segmentation is only supported for inference in this repo.

[SofianeCordon]

Segmentation training is not implemented, there is no segmentation loss if you look for it, dataloader does not support segmentation, neither do the augmentations. Segmentation is only supported for inference in this repo.

Thanks