🚀 gaunernst/vit_small_patch8_gap_112.cosface_ms1mv3

这是一个用于人脸识别的视觉变换器（ViT）模型，在MS1MV3数据集上进行训练。该模型与timm库完全兼容，可助力人脸识别相关任务。

🚀 快速开始

此模型为用于人脸识别的视觉变换器（ViT），在MS1MV3数据集上完成训练。训练代码可参考该仓库：https://github.com/gau-nernst/timm-face 。

💻 使用示例

基础用法

import timm
import torch.nn.functional as F

model = timm.create_model("hf_hub:gaunernst/vit_small_patch8_gap_112.cosface_ms1mv3", pretrained=True).eval()
embs = model(torch.randn(1, 3, 112, 112))  # output shape (1, 512)
embs = F.normalize(embs, dim=1)  # model output is not normalized

📚 详细文档