I really need the code for downstream segmentation tasks using MAE pre-training weights

lightly-ai / lightly

A python library for self-supervised learning on images.

https://docs.lightly.ai/self-supervised-learning/

MIT License

2.92k stars 248 forks source link

I really need the code for downstream segmentation tasks using MAE pre-training weights #1591

Closed rodmarn closed 3 weeks ago

guarin commented 1 month ago

Hi! We sadly don't have example code for segmentation finetuning using MAE. AFAIK this code was also not published in the original MAE repo. The paper states that for segmentation the code from https://github.com/microsoft/unilm/tree/master/beit was used.

May I ask what your use-case is? Are you looking to finetune MAE using a specific segmentation library or are you interested in reproducing the results from the paper?

guarin commented 3 weeks ago

Will close this for now, please reopen if you have more questions.