/seg-dinov2

Fine-tuning dino v2 for semantic segmentation task on MSCOCO.

Primary LanguagePythonOtherNOASSERTION

seg-dinov2

Fine-tuning dino v2 for semantic segmentation task on MSCOCO.

Implementation Detail

Backbone

  • dinov2-vitb/14 as backbone

Head

  • Linear layer + conv layer

  • variation of linear tuning (refer to section 7.4 of Dinov2)

Data preparation code for segmentation task

Trained on

RTX3090*2 with batch size of 8.

Prediction

Prediction

GT