Train Image Segmentation Models

[AnikaP]

Hi, thanks for the work you're doing!
I'm currently looking into image segmentation models (like eva) and have a hard time understanding the training pipeline (using the train.py script). In my understanding the dataloader and train script are designated for classification purposes? Any input how to train for segmentation to help me get started would be appreciated!

[hankyul2]

Hi @AnikaP

How about starting with MMSegmentation library? They support various Timm models as backbone so that you can freely change the backbone network architecture as your purpose.

Thank you.

Hankyul.