Is there any other pre-trained model in this pkg besides segformer?

[Student2Pro]

Hi, I'm starting a project in semantic segmentation.
However, due to limited hardware, I have to use pre-trained models instead of training them myself.
I would be appreciated if anyone can answer my question.

[qubvel]

Hey, at the moment only segformer is integrated as fully-pretrained model. The library is more oriented to fine-tuning partially pertained models.

In case you need pertained segmentation models you can find several ones on Hugging Face Hub

https://huggingface.co/models?pipeline_tag=image-segmentation&sort=trending

[Student2Pro]

@qubvel Thanks for your help, have a nice day!