dinov2-small facebook

类型
model
任务
image-feature-extraction
框架
transformers
下载量
4,664,473
点赞
71
访问
public
文件
0

标签

  • 视觉模型
  • 图像特征提取
  • 图像分类
  • 计算机视觉
  • 自监督学习
  • 深度学习
  • transformers
  • PyTorch

摘要

DINOv2-small是一个使用自监督方法训练的视觉Transformer(ViT)模型,无需标注数据即可学习鲁棒的图像视觉特征。该模型不包含任何微调头,输出图像的通用表示,可用于图像特征提取并作为下游任务(如图像分类、检索等)的基础编码器。适用于需要在预训练编码器上加线性层完成分类、语义检索等视觉任务的场景。

README

--- license: apache-2.0 tags: - dino - vision --- # 使用 DINOv2 训练的 Vision Transformer(小型模型) 使用 DINOv2 方法训练的 Vision Transformer(ViT)模型。该模型在 Oquab 等人发表的论文 [DINOv2: Learning Robust Visual Features without Supervision](https://arxiv.org/abs/2304.07193) 中提出,并首次发布于[此代码库](https://github.com/facebookresearc…

查看完整页面 · 查看原文