数据集:

• gaunernst/ms1mv3-recordio 库名称: timm 标签:
• 图像特征提取
• timm

模型卡片 gaunernst/vit_tiny_patch8_112.arcface_ms1mv3

一个用于人脸识别的视觉变换器(ViT)模型，基于MS1MV3数据集训练。该模型使用此代码库训练: https://github.com/gau-nernst/timm-face。完全兼容timm库。

使用方式

import timm
import torch.nn.functional as F

model = timm.create_model("hf_hub:gaunernst/vit_tiny_patch8_112.arcface_ms1mv3", pretrained=True).eval()
embs = model(torch.randn(1, 3, 112, 112))  # 输出形状 (1, 512)
embs = F.normalize(embs, dim=1)  # 模型输出未归一化