
dino
facebookresearch
PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
AI 简介
DINO 是一个基于 PyTorch 的自监督学习方法,用于训练视觉变换器(Vision Transformers)。项目提供了预训练模型和详细的实现代码,支持多种架构如 ViT-S/16、ViT-S/8 和 ViT-B/16,并且可以下载仅包含骨干网络权重的版本或完整的检查点。其核心技术特点是通过自监督学习使得模型在不需要大量标注数据的情况下也能达到较好的性能表现。适用于需要高效利用未标记图像数据进行预训练,进而提升下游任务效果的各种计算机视觉场景中,比如图像分类、目标检测等。
Python
Apache License 2.07.6k
Stars
1k
Forks
1
Watchers
101
Issues
Star 增长
今日+5
近 7 天+12
近 30 天+38
综合评分84.86
默认分支main