dinov2 icon indicating copy to clipboard operation
dinov2 copied to clipboard

Training pipeline of linear probing for semantic segmentation and depth estimation

Open ywyue opened this issue 5 months ago • 1 comments

Dear DINOv2 team, thank you for this amazing work! If I am correct, I only found the whole pipeline of linear probing for classification on ImageNet in https://github.com/facebookresearch/dinov2/blob/main/dinov2/eval/linear.py.

Is it possible for you to also release the training pipeline of linear probing for semantic segmentation (ADE20k, Cityscapes, Pascal VOC 2012) and depth estimation (NYU-Depth V2, KITTI, SUN-RGBD), including dataloader, data augmentation used, training schedule, etc? I believe this will serve as a standard testbed for follow-up works to compare with DINOv2 in a fair setup (i.e. excluding the effect of data augmentation, training schedule, etc). Thank you very much!

ywyue avatar Jan 11 '24 23:01 ywyue