facebookresearch/dinov2
PyTorch code and models for the DINOv2 self-supervised learning method.
解决的问题
DINOv2 提供强大且高性能的视觉特征,可用于各种计算机视觉任务,而无需进行大量的微调。它通过在没有监督的情况下从图像中学习视觉表征,消除了对大规模标注数据集的需求。
运作原理
它使用在 1.42 亿张图像的庞大数据集上预训练的 Vision Transformer (ViT) 架构,且无需任何标签或注释。该项目提供预训练的骨干网络(包括带有「registers」以提高特征质量的版本),以及用于图像分类、深度估计和语义分割等任务的专用头部。它还包括对特定领域骨干网络的支持,例如用于医学影像的 XRay-DINO 和用于显微镜检查的 Cell-DINO。
目标对象
需要高质量、通用视觉特征以进行分类、深度估计和语义分割等下游任务的计算机视觉研究人员和开发人员。
特色亮点
- 自我监督学习:在 1.42 亿张图像上进行预训练,无需标签。
- 多功能骨干网络:提供多种 ViT 尺寸(Small, Base, Large, Giant)以及带有 registers 的版本。
- 支持多种任务:提供用于图像分类、深度估计 (NYUd, KITTI) 和语义分割 (ADE20K, VOC2012) 的预训练头部。
- 领域扩展:包括对 X 光分析和细胞荧光显微镜检查的专门支持。
- 视觉-语言对齐:包含
dino.txt,用于通过图像和像素级对齐进行零样本任务。
相关
- 项目
- 项目
- 项目
- 项目