数据集:

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

gaunernst/vit_tiny_patch8_112.cosface_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.cosface_ms1mv3", pretrained=True).eval()
embs = model(torch.randn(1, 3, 112, 112))  # 输出形状为 (1, 512)
embs = F.normalize(embs, dim=1)  # 模型输出未归一化