search

lightly-ai / lightly

A python library for self-supervised learning on images.
https://docs.lightly.ai/self-supervised-learning/
MIT License
3.17k stars 285 forks source link

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

Closed rodmarn closed 3 months ago

guarin commented 4 months 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 months ago

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